├── .github └── workflows │ ├── python-publish.yml │ └── python-test.yml ├── .gitignore ├── .readthedocs.yml ├── LICENSE.md ├── MANIFEST.in ├── README.md ├── codecov.yml ├── dev-requirements.in ├── dev-requirements.txt ├── docs ├── Makefile └── source │ ├── conf.py │ ├── datasets.rst │ ├── evaluation.rst │ ├── getting-started.rst │ ├── index.rst │ ├── loss.rst │ └── references.bib ├── examples └── 01-basic-usage.py ├── pyproject.toml ├── pytorchltr ├── __init__.py ├── click_simulation │ ├── __init__.py │ └── pbm.py ├── datasets │ ├── __init__.py │ ├── list_sampler.py │ └── svmrank │ │ ├── __init__.py │ │ ├── example3.py │ │ ├── istella.py │ │ ├── istella_s.py │ │ ├── istella_x.py │ │ ├── mslr10k.py │ │ ├── mslr30k.py │ │ ├── parser │ │ ├── .gitignore │ │ ├── __init__.py │ │ ├── svmrank_parser.h │ │ └── svmrank_parser.pyx │ │ └── svmrank.py ├── evaluation │ ├── __init__.py │ ├── arp.py │ ├── dcg.py │ └── trec.py ├── loss │ ├── __init__.py │ ├── pairwise_additive.py │ └── pairwise_lambda.py └── utils │ ├── __init__.py │ ├── downloader.py │ ├── file.py │ ├── progress.py │ └── tensor_operations.py ├── requirements.in ├── requirements.txt ├── setup.py └── tests ├── __init__.py ├── click_simulation ├── __init__.py └── test_pbm.py ├── datasets ├── __init__.py ├── resources │ └── dataset.txt ├── svmrank │ ├── __init__.py │ ├── test_example3.py │ ├── test_istella.py │ ├── test_istella_s.py │ ├── test_istella_x.py │ ├── test_mslr10k.py │ ├── test_mslr30k.py │ └── test_svmrank.py └── test_list_sampler.py ├── evaluation ├── __init__.py ├── test_arp.py ├── test_dcg.py └── test_trec.py ├── loss ├── __init__.py ├── test_pairwise_additive.py └── test_pairwise_lambda.py ├── test_integration.py └── utils ├── __init__.py ├── test_downloader.py ├── test_file.py └── test_tensor_operations.py /.github/workflows/python-publish.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rjagerman/pytorchltr/HEAD/.github/workflows/python-publish.yml -------------------------------------------------------------------------------- /.github/workflows/python-test.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rjagerman/pytorchltr/HEAD/.github/workflows/python-test.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rjagerman/pytorchltr/HEAD/.gitignore -------------------------------------------------------------------------------- /.readthedocs.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rjagerman/pytorchltr/HEAD/.readthedocs.yml -------------------------------------------------------------------------------- /LICENSE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rjagerman/pytorchltr/HEAD/LICENSE.md -------------------------------------------------------------------------------- /MANIFEST.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rjagerman/pytorchltr/HEAD/MANIFEST.in -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rjagerman/pytorchltr/HEAD/README.md -------------------------------------------------------------------------------- /codecov.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rjagerman/pytorchltr/HEAD/codecov.yml -------------------------------------------------------------------------------- /dev-requirements.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rjagerman/pytorchltr/HEAD/dev-requirements.in -------------------------------------------------------------------------------- /dev-requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rjagerman/pytorchltr/HEAD/dev-requirements.txt -------------------------------------------------------------------------------- /docs/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rjagerman/pytorchltr/HEAD/docs/Makefile -------------------------------------------------------------------------------- /docs/source/conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rjagerman/pytorchltr/HEAD/docs/source/conf.py -------------------------------------------------------------------------------- /docs/source/datasets.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rjagerman/pytorchltr/HEAD/docs/source/datasets.rst -------------------------------------------------------------------------------- /docs/source/evaluation.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rjagerman/pytorchltr/HEAD/docs/source/evaluation.rst -------------------------------------------------------------------------------- /docs/source/getting-started.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rjagerman/pytorchltr/HEAD/docs/source/getting-started.rst -------------------------------------------------------------------------------- /docs/source/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rjagerman/pytorchltr/HEAD/docs/source/index.rst -------------------------------------------------------------------------------- /docs/source/loss.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rjagerman/pytorchltr/HEAD/docs/source/loss.rst -------------------------------------------------------------------------------- /docs/source/references.bib: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rjagerman/pytorchltr/HEAD/docs/source/references.bib -------------------------------------------------------------------------------- /examples/01-basic-usage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rjagerman/pytorchltr/HEAD/examples/01-basic-usage.py -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rjagerman/pytorchltr/HEAD/pyproject.toml -------------------------------------------------------------------------------- /pytorchltr/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rjagerman/pytorchltr/HEAD/pytorchltr/__init__.py -------------------------------------------------------------------------------- /pytorchltr/click_simulation/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /pytorchltr/click_simulation/pbm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rjagerman/pytorchltr/HEAD/pytorchltr/click_simulation/pbm.py -------------------------------------------------------------------------------- /pytorchltr/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rjagerman/pytorchltr/HEAD/pytorchltr/datasets/__init__.py -------------------------------------------------------------------------------- /pytorchltr/datasets/list_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rjagerman/pytorchltr/HEAD/pytorchltr/datasets/list_sampler.py -------------------------------------------------------------------------------- /pytorchltr/datasets/svmrank/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /pytorchltr/datasets/svmrank/example3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rjagerman/pytorchltr/HEAD/pytorchltr/datasets/svmrank/example3.py -------------------------------------------------------------------------------- /pytorchltr/datasets/svmrank/istella.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rjagerman/pytorchltr/HEAD/pytorchltr/datasets/svmrank/istella.py -------------------------------------------------------------------------------- /pytorchltr/datasets/svmrank/istella_s.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rjagerman/pytorchltr/HEAD/pytorchltr/datasets/svmrank/istella_s.py -------------------------------------------------------------------------------- /pytorchltr/datasets/svmrank/istella_x.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rjagerman/pytorchltr/HEAD/pytorchltr/datasets/svmrank/istella_x.py -------------------------------------------------------------------------------- /pytorchltr/datasets/svmrank/mslr10k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rjagerman/pytorchltr/HEAD/pytorchltr/datasets/svmrank/mslr10k.py -------------------------------------------------------------------------------- /pytorchltr/datasets/svmrank/mslr30k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rjagerman/pytorchltr/HEAD/pytorchltr/datasets/svmrank/mslr30k.py -------------------------------------------------------------------------------- /pytorchltr/datasets/svmrank/parser/.gitignore: -------------------------------------------------------------------------------- 1 | # Generated by cython 2 | svmrank_parser.c -------------------------------------------------------------------------------- /pytorchltr/datasets/svmrank/parser/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rjagerman/pytorchltr/HEAD/pytorchltr/datasets/svmrank/parser/__init__.py -------------------------------------------------------------------------------- /pytorchltr/datasets/svmrank/parser/svmrank_parser.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rjagerman/pytorchltr/HEAD/pytorchltr/datasets/svmrank/parser/svmrank_parser.h -------------------------------------------------------------------------------- /pytorchltr/datasets/svmrank/parser/svmrank_parser.pyx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rjagerman/pytorchltr/HEAD/pytorchltr/datasets/svmrank/parser/svmrank_parser.pyx -------------------------------------------------------------------------------- /pytorchltr/datasets/svmrank/svmrank.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rjagerman/pytorchltr/HEAD/pytorchltr/datasets/svmrank/svmrank.py -------------------------------------------------------------------------------- /pytorchltr/evaluation/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rjagerman/pytorchltr/HEAD/pytorchltr/evaluation/__init__.py -------------------------------------------------------------------------------- /pytorchltr/evaluation/arp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rjagerman/pytorchltr/HEAD/pytorchltr/evaluation/arp.py -------------------------------------------------------------------------------- /pytorchltr/evaluation/dcg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rjagerman/pytorchltr/HEAD/pytorchltr/evaluation/dcg.py -------------------------------------------------------------------------------- /pytorchltr/evaluation/trec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rjagerman/pytorchltr/HEAD/pytorchltr/evaluation/trec.py -------------------------------------------------------------------------------- /pytorchltr/loss/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rjagerman/pytorchltr/HEAD/pytorchltr/loss/__init__.py -------------------------------------------------------------------------------- /pytorchltr/loss/pairwise_additive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rjagerman/pytorchltr/HEAD/pytorchltr/loss/pairwise_additive.py -------------------------------------------------------------------------------- /pytorchltr/loss/pairwise_lambda.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rjagerman/pytorchltr/HEAD/pytorchltr/loss/pairwise_lambda.py -------------------------------------------------------------------------------- /pytorchltr/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rjagerman/pytorchltr/HEAD/pytorchltr/utils/__init__.py -------------------------------------------------------------------------------- /pytorchltr/utils/downloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rjagerman/pytorchltr/HEAD/pytorchltr/utils/downloader.py -------------------------------------------------------------------------------- /pytorchltr/utils/file.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rjagerman/pytorchltr/HEAD/pytorchltr/utils/file.py -------------------------------------------------------------------------------- /pytorchltr/utils/progress.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rjagerman/pytorchltr/HEAD/pytorchltr/utils/progress.py -------------------------------------------------------------------------------- /pytorchltr/utils/tensor_operations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rjagerman/pytorchltr/HEAD/pytorchltr/utils/tensor_operations.py -------------------------------------------------------------------------------- /requirements.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rjagerman/pytorchltr/HEAD/requirements.in -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rjagerman/pytorchltr/HEAD/requirements.txt -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rjagerman/pytorchltr/HEAD/setup.py -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/click_simulation/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/click_simulation/test_pbm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rjagerman/pytorchltr/HEAD/tests/click_simulation/test_pbm.py -------------------------------------------------------------------------------- /tests/datasets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/datasets/resources/dataset.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rjagerman/pytorchltr/HEAD/tests/datasets/resources/dataset.txt -------------------------------------------------------------------------------- /tests/datasets/svmrank/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/datasets/svmrank/test_example3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rjagerman/pytorchltr/HEAD/tests/datasets/svmrank/test_example3.py -------------------------------------------------------------------------------- /tests/datasets/svmrank/test_istella.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rjagerman/pytorchltr/HEAD/tests/datasets/svmrank/test_istella.py -------------------------------------------------------------------------------- /tests/datasets/svmrank/test_istella_s.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rjagerman/pytorchltr/HEAD/tests/datasets/svmrank/test_istella_s.py -------------------------------------------------------------------------------- /tests/datasets/svmrank/test_istella_x.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rjagerman/pytorchltr/HEAD/tests/datasets/svmrank/test_istella_x.py -------------------------------------------------------------------------------- /tests/datasets/svmrank/test_mslr10k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rjagerman/pytorchltr/HEAD/tests/datasets/svmrank/test_mslr10k.py -------------------------------------------------------------------------------- /tests/datasets/svmrank/test_mslr30k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rjagerman/pytorchltr/HEAD/tests/datasets/svmrank/test_mslr30k.py -------------------------------------------------------------------------------- /tests/datasets/svmrank/test_svmrank.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rjagerman/pytorchltr/HEAD/tests/datasets/svmrank/test_svmrank.py -------------------------------------------------------------------------------- /tests/datasets/test_list_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rjagerman/pytorchltr/HEAD/tests/datasets/test_list_sampler.py -------------------------------------------------------------------------------- /tests/evaluation/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/evaluation/test_arp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rjagerman/pytorchltr/HEAD/tests/evaluation/test_arp.py -------------------------------------------------------------------------------- /tests/evaluation/test_dcg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rjagerman/pytorchltr/HEAD/tests/evaluation/test_dcg.py -------------------------------------------------------------------------------- /tests/evaluation/test_trec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rjagerman/pytorchltr/HEAD/tests/evaluation/test_trec.py -------------------------------------------------------------------------------- /tests/loss/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/loss/test_pairwise_additive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rjagerman/pytorchltr/HEAD/tests/loss/test_pairwise_additive.py -------------------------------------------------------------------------------- /tests/loss/test_pairwise_lambda.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rjagerman/pytorchltr/HEAD/tests/loss/test_pairwise_lambda.py -------------------------------------------------------------------------------- /tests/test_integration.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rjagerman/pytorchltr/HEAD/tests/test_integration.py -------------------------------------------------------------------------------- /tests/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/utils/test_downloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rjagerman/pytorchltr/HEAD/tests/utils/test_downloader.py -------------------------------------------------------------------------------- /tests/utils/test_file.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rjagerman/pytorchltr/HEAD/tests/utils/test_file.py -------------------------------------------------------------------------------- /tests/utils/test_tensor_operations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rjagerman/pytorchltr/HEAD/tests/utils/test_tensor_operations.py --------------------------------------------------------------------------------