pytorch/ignite
Improve distributed tests for Confusion Matrix (CM and MLCM)
オープン
#1,657 opened on 2021/02/20
HacktoberfestPyDataGlobalenhancementhelp wantedmodule: metrics
Repository metrics
- Stars
- (4,313 個のスター)
- PR merge metrics
- (PR metrics pending)
説明
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