├── .gitignore ├── .pre-commit-config.yaml ├── LICENSE.md ├── README.md ├── mypy.ini ├── poetry.lock ├── pyproject.toml ├── ranking_metrics_torch ├── __init__.py ├── avg_precision.py ├── common.py ├── cumulative_gain.py └── precision_recall.py └── tests ├── __init__.py ├── conftest.py ├── test_avg_precision.py ├── test_common.py ├── test_cumulative_gain.py └── test_precision_recall.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/karlhigley/ranking-metrics-torch/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/karlhigley/ranking-metrics-torch/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /LICENSE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/karlhigley/ranking-metrics-torch/HEAD/LICENSE.md -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/karlhigley/ranking-metrics-torch/HEAD/README.md -------------------------------------------------------------------------------- /mypy.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/karlhigley/ranking-metrics-torch/HEAD/mypy.ini -------------------------------------------------------------------------------- /poetry.lock: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/karlhigley/ranking-metrics-torch/HEAD/poetry.lock -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/karlhigley/ranking-metrics-torch/HEAD/pyproject.toml -------------------------------------------------------------------------------- /ranking_metrics_torch/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ranking_metrics_torch/avg_precision.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/karlhigley/ranking-metrics-torch/HEAD/ranking_metrics_torch/avg_precision.py -------------------------------------------------------------------------------- /ranking_metrics_torch/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/karlhigley/ranking-metrics-torch/HEAD/ranking_metrics_torch/common.py -------------------------------------------------------------------------------- /ranking_metrics_torch/cumulative_gain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/karlhigley/ranking-metrics-torch/HEAD/ranking_metrics_torch/cumulative_gain.py -------------------------------------------------------------------------------- /ranking_metrics_torch/precision_recall.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/karlhigley/ranking-metrics-torch/HEAD/ranking_metrics_torch/precision_recall.py -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/karlhigley/ranking-metrics-torch/HEAD/tests/conftest.py -------------------------------------------------------------------------------- /tests/test_avg_precision.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/karlhigley/ranking-metrics-torch/HEAD/tests/test_avg_precision.py -------------------------------------------------------------------------------- /tests/test_common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/karlhigley/ranking-metrics-torch/HEAD/tests/test_common.py -------------------------------------------------------------------------------- /tests/test_cumulative_gain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/karlhigley/ranking-metrics-torch/HEAD/tests/test_cumulative_gain.py -------------------------------------------------------------------------------- /tests/test_precision_recall.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/karlhigley/ranking-metrics-torch/HEAD/tests/test_precision_recall.py --------------------------------------------------------------------------------