├── .gitignore ├── LICENSE.md ├── README.md ├── requirements.txt └── src ├── __init__.py ├── main ├── EASE.py ├── __init__.py ├── metrics.py └── train.py ├── notebooks ├── TorchEASE_GoodReads.ipynb └── __init__.py └── test ├── .pytest_cache ├── README.md └── v │ └── cache │ ├── lastfailed │ └── nodeids ├── __init__.py ├── test_ease.py └── test_hitrate.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/franckjay/TorchEASE/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/franckjay/TorchEASE/HEAD/LICENSE.md -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/franckjay/TorchEASE/HEAD/README.md -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | torch 1.7.1 2 | pandas 0.25.3 -------------------------------------------------------------------------------- /src/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/main/EASE.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/franckjay/TorchEASE/HEAD/src/main/EASE.py -------------------------------------------------------------------------------- /src/main/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/main/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/franckjay/TorchEASE/HEAD/src/main/metrics.py -------------------------------------------------------------------------------- /src/main/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/franckjay/TorchEASE/HEAD/src/main/train.py -------------------------------------------------------------------------------- /src/notebooks/TorchEASE_GoodReads.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/franckjay/TorchEASE/HEAD/src/notebooks/TorchEASE_GoodReads.ipynb -------------------------------------------------------------------------------- /src/notebooks/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/test/.pytest_cache/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/franckjay/TorchEASE/HEAD/src/test/.pytest_cache/README.md -------------------------------------------------------------------------------- /src/test/.pytest_cache/v/cache/lastfailed: -------------------------------------------------------------------------------- 1 | {} -------------------------------------------------------------------------------- /src/test/.pytest_cache/v/cache/nodeids: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/franckjay/TorchEASE/HEAD/src/test/.pytest_cache/v/cache/nodeids -------------------------------------------------------------------------------- /src/test/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/test/test_ease.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/franckjay/TorchEASE/HEAD/src/test/test_ease.py -------------------------------------------------------------------------------- /src/test/test_hitrate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/franckjay/TorchEASE/HEAD/src/test/test_hitrate.py --------------------------------------------------------------------------------