lightning/pytorch_lightning
Tadej Svetina c8f605e47d
Classification metrics overhaul: precision & recall (4/n) (#4842)
* Add stuff

* Change metrics documentation layout

* Add stuff

* Add stat scores

* Change testing utils

* Replace len(*.shape) with *.ndim

* More descriptive error message for input formatting

* Replace movedim with permute

* PEP 8 compliance

* WIP

* Add reduce_scores function

* Temporarily add back legacy class_reduce

* Division with float

* PEP 8 compliance

* Remove precision recall

* Replace movedim with permute

* Add back tests

* Add empty newlines

* Add precision recall back

* Add empty line

* Fix permute

* Fix some issues with old versions of PyTorch

* Style changes in error messages

* More error message style improvements

* Fix typo in docs

* Add more descriptive variable names in utils

* Change internal var names

* Revert unwanted changes

* Revert unwanted changes pt 2

* Update metrics interface

* Add top_k parameter

* Add back reduce function

* Add stuff

* PEP3

* Add depreciation

* PEP8

* Deprecate param

* PEP8

* Fix and simplify testing for older PT versions

* Update Changelog

* Remove redundant import

* Add tests to increase coverage

* Remove zero_division

* fix zero_division

* Add zero_div + edge case tests

* Reorder cls metric args

* Add back quotes for is_multiclass

* Add precision_recall and tests

* PEP8

* Fix docs

* Fix docs

* Update

* Change precision_recall output

* PEP8/isort

* Add method _get_final_stats

* Fix depr test

* Add comment to deprecation tests

* isort

* Apply suggestions from code review

Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>

* Add typing to test

* Add matc str to pytest.raises

Co-authored-by: chaton <thomas@grid.ai>
Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
2021-01-18 03:24:13 -05:00
..
accelerators Fix pre-commit isort failure on pytorch_lightning/accelerators (#5503) 2021-01-16 14:10:56 -05:00
callbacks Fix isort a few failures (#5504) 2021-01-15 17:44:27 -05:00
cluster_environments Fix isort a few failures (#5504) 2021-01-15 17:44:27 -05:00
core Refactor LightningDistributedDataParallel (#5185) 2021-01-13 14:35:42 -05:00
distributed fix/enable - check F401 (#5201) 2020-12-21 10:15:04 +01:00
loggers Fix isort failures in loggers (#5527) 2021-01-15 22:53:56 +05:30
metrics Classification metrics overhaul: precision & recall (4/n) (#4842) 2021-01-18 03:24:13 -05:00
overrides Refactor LightningDistributedDataParallel (#5185) 2021-01-13 14:35:42 -05:00
plugins Refactor LightningDistributedDataParallel (#5185) 2021-01-13 14:35:42 -05:00
profiler Fix isort a few failures (#5504) 2021-01-15 17:44:27 -05:00
trainer Refactor LightningDistributedDataParallel (#5185) 2021-01-13 14:35:42 -05:00
tuner Fix isort a few failures (#5504) 2021-01-15 17:44:27 -05:00
utilities Fix isort failures in utilities (#5530) 2021-01-15 13:57:40 -05:00
__init__.py flake8 ++ 2021-01-05 09:58:37 +01:00
py.typed make PyTorch Lightning PEP 561 Compliant (#3187) 2020-09-09 13:37:03 -04:00
setup_tools.py update isort config (#5335) 2021-01-06 12:49:23 +01:00