pytorch/ignite
GitHub で見るImprove distributed tests for Confusion Matrix (CM and MLCM)
Open
#1,657 opened on 2021年2月20日
HacktoberfestPyDataGlobalenhancementhelp wantedmodule: metrics
説明
Currently, the testing data is generated without taking into account distributed context: https://github.com/pytorch/ignite/blob/e17acc740d38909ddbbed6648423750abe593e58/tests/ignite/metrics/test_confusion_matrix.py#L559-L565
same for MultiLabelConfusionMatrix tests.
We would like to rewrite the tests in a similar way as here : https://github.com/pytorch/ignite/blob/e17acc740d38909ddbbed6648423750abe593e58/tests/ignite/metrics/test_recall.py#L778-L785
See https://github.com/pytorch/ignite/issues#issuecomment-782750122
cc @touqir14