lightning/pytorch_lightning/metrics/functional
Cookie_thief a552d4a2d5
fix normalize mode at confusion matrix (replace nans with zeros) (#3465)
* replace nans to 0 at conf. matrix & update tests

* cm.isnan() -> torch.isnan(cm)

* fix row-wise division while normalize

* update tests

* pep8 fix

* Update tests/metrics/test_classification.py

add comment to test

Co-authored-by: Nicki Skafte <skaftenicki@gmail.com>

* Update tests/metrics/functional/test_classification.py

Co-authored-by: Nicki Skafte <skaftenicki@gmail.com>

* Update pytorch_lightning/metrics/functional/classification.py

Co-authored-by: Nicki Skafte <skaftenicki@gmail.com>

* final update

Co-authored-by: Nicki Skafte <skaftenicki@gmail.com>
2020-09-14 10:05:51 +02:00
..
__init__.py added emb similarity (#3349) 2020-09-04 09:45:21 -04:00
classification.py fix normalize mode at confusion matrix (replace nans with zeros) (#3465) 2020-09-14 10:05:51 +02:00
nlp.py fix reduction docstring and clean tests (#2885) 2020-08-09 06:03:24 -04:00
reduction.py Native torch metrics (#1488) 2020-06-13 08:47:25 -04:00
regression.py Fix RMSLE metric (#3188) 2020-08-26 08:02:53 -04:00
self_supervised.py [Metrics] class based embedding similarity + tests (#3358) 2020-09-11 12:11:50 +02:00