lightning/pytorch_lightning/metrics
Nicki Skafte 93cf6d0054
[Metrics] class based embedding similarity + tests (#3358)
* embedding similarity class + test

* fix tests

* fix pep8

* add docs

* noindex

* Update docs/source/metrics.rst

* Update pytorch_lightning/metrics/self_supervised.py

Co-authored-by: Rohit Gupta <rohitgr1998@gmail.com>

* Update pytorch_lightning/metrics/self_supervised.py

Co-authored-by: Rohit Gupta <rohitgr1998@gmail.com>

* suggestions

* changes to init

* move __all__

* fix imports

* Apply suggestions from code review

* assert typo

* change import

Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>
Co-authored-by: Justus Schock <12886177+justusschock@users.noreply.github.com>
Co-authored-by: Rohit Gupta <rohitgr1998@gmail.com>
Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
Co-authored-by: Nicki Skafte <nugginea@gmail.com>
2020-09-11 12:11:50 +02:00
..
functional [Metrics] class based embedding similarity + tests (#3358) 2020-09-11 12:11:50 +02:00
__init__.py [Metrics] class based embedding similarity + tests (#3358) 2020-09-11 12:11:50 +02:00
classification.py add num_classes argument to confusion matrix (#3450) 2020-09-10 18:39:04 -04:00
converters.py Fix: gather_all_tensors cross GPUs in DDP (#3319) 2020-09-03 12:27:32 +02:00
metric.py New modular metric interface (#2528) 2020-08-26 13:01:29 +02:00
nlp.py added copyright notices (#3062) 2020-08-19 22:03:22 -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
sklearns.py added copyright notices (#3062) 2020-08-19 22:03:22 -04:00