pytorch/ignite
Improve distributed tests for Confusion Matrix (CM and MLCM)
Ouverte
#1 657 ouverte le 20 févr. 2021
HacktoberfestPyDataGlobalenhancementhelp wantedmodule: metrics
Métriques du dépôt
- Stars
- (4 313 étoiles)
- Métriques de merge PR
- (Métriques PR en attente)
Description
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