├── .gitignore ├── README.md ├── configs ├── amzb.json ├── lfm1b.json └── ml1m.json ├── data_misc └── kcore_interactions.py ├── figures ├── mojito.pdf └── mojito.png ├── mojito ├── __init__.py ├── __main__.py ├── commands │ ├── __init__.py │ ├── eval.py │ └── train.py ├── configuration.py ├── data │ ├── __init__.py │ ├── datasets │ │ ├── __init__.py │ │ ├── dataset.py │ │ └── tempo │ │ │ ├── __init__.py │ │ │ └── moji.py │ ├── loaders │ │ ├── __init__.py │ │ ├── loader.py │ │ ├── test_loader.py │ │ └── train_loader.py │ ├── samplers │ │ ├── __init__.py │ │ ├── mojito_sampler.py │ │ └── sampler.py │ └── splitter.py ├── eval │ ├── __init__.py │ └── evaluator.py ├── logging.py ├── models │ ├── __init__.py │ ├── abs │ │ ├── __init__.py │ │ ├── sas.py │ │ └── seq.py │ ├── core │ │ ├── __init__.py │ │ ├── fish.py │ │ └── net.py │ ├── model.py │ └── net.py ├── train │ ├── __init__.py │ └── trainer.py └── utils │ ├── __init__.py │ ├── params.py │ └── tempo.py └── script └── run_mojito.sh /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deezer/sigir23-mojito/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deezer/sigir23-mojito/HEAD/README.md -------------------------------------------------------------------------------- /configs/amzb.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deezer/sigir23-mojito/HEAD/configs/amzb.json -------------------------------------------------------------------------------- /configs/lfm1b.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deezer/sigir23-mojito/HEAD/configs/lfm1b.json -------------------------------------------------------------------------------- /configs/ml1m.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deezer/sigir23-mojito/HEAD/configs/ml1m.json -------------------------------------------------------------------------------- /data_misc/kcore_interactions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deezer/sigir23-mojito/HEAD/data_misc/kcore_interactions.py -------------------------------------------------------------------------------- /figures/mojito.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deezer/sigir23-mojito/HEAD/figures/mojito.pdf -------------------------------------------------------------------------------- /figures/mojito.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deezer/sigir23-mojito/HEAD/figures/mojito.png -------------------------------------------------------------------------------- /mojito/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deezer/sigir23-mojito/HEAD/mojito/__init__.py -------------------------------------------------------------------------------- /mojito/__main__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deezer/sigir23-mojito/HEAD/mojito/__main__.py -------------------------------------------------------------------------------- /mojito/commands/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deezer/sigir23-mojito/HEAD/mojito/commands/__init__.py -------------------------------------------------------------------------------- /mojito/commands/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deezer/sigir23-mojito/HEAD/mojito/commands/eval.py -------------------------------------------------------------------------------- /mojito/commands/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deezer/sigir23-mojito/HEAD/mojito/commands/train.py -------------------------------------------------------------------------------- /mojito/configuration.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deezer/sigir23-mojito/HEAD/mojito/configuration.py -------------------------------------------------------------------------------- /mojito/data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /mojito/data/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deezer/sigir23-mojito/HEAD/mojito/data/datasets/__init__.py -------------------------------------------------------------------------------- /mojito/data/datasets/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deezer/sigir23-mojito/HEAD/mojito/data/datasets/dataset.py -------------------------------------------------------------------------------- /mojito/data/datasets/tempo/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deezer/sigir23-mojito/HEAD/mojito/data/datasets/tempo/__init__.py -------------------------------------------------------------------------------- /mojito/data/datasets/tempo/moji.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deezer/sigir23-mojito/HEAD/mojito/data/datasets/tempo/moji.py -------------------------------------------------------------------------------- /mojito/data/loaders/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deezer/sigir23-mojito/HEAD/mojito/data/loaders/__init__.py -------------------------------------------------------------------------------- /mojito/data/loaders/loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deezer/sigir23-mojito/HEAD/mojito/data/loaders/loader.py -------------------------------------------------------------------------------- /mojito/data/loaders/test_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deezer/sigir23-mojito/HEAD/mojito/data/loaders/test_loader.py -------------------------------------------------------------------------------- /mojito/data/loaders/train_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deezer/sigir23-mojito/HEAD/mojito/data/loaders/train_loader.py -------------------------------------------------------------------------------- /mojito/data/samplers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deezer/sigir23-mojito/HEAD/mojito/data/samplers/__init__.py -------------------------------------------------------------------------------- /mojito/data/samplers/mojito_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deezer/sigir23-mojito/HEAD/mojito/data/samplers/mojito_sampler.py -------------------------------------------------------------------------------- /mojito/data/samplers/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deezer/sigir23-mojito/HEAD/mojito/data/samplers/sampler.py -------------------------------------------------------------------------------- /mojito/data/splitter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deezer/sigir23-mojito/HEAD/mojito/data/splitter.py -------------------------------------------------------------------------------- /mojito/eval/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /mojito/eval/evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deezer/sigir23-mojito/HEAD/mojito/eval/evaluator.py -------------------------------------------------------------------------------- /mojito/logging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deezer/sigir23-mojito/HEAD/mojito/logging.py -------------------------------------------------------------------------------- /mojito/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deezer/sigir23-mojito/HEAD/mojito/models/__init__.py -------------------------------------------------------------------------------- /mojito/models/abs/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /mojito/models/abs/sas.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deezer/sigir23-mojito/HEAD/mojito/models/abs/sas.py -------------------------------------------------------------------------------- /mojito/models/abs/seq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deezer/sigir23-mojito/HEAD/mojito/models/abs/seq.py -------------------------------------------------------------------------------- /mojito/models/core/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deezer/sigir23-mojito/HEAD/mojito/models/core/__init__.py -------------------------------------------------------------------------------- /mojito/models/core/fish.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deezer/sigir23-mojito/HEAD/mojito/models/core/fish.py -------------------------------------------------------------------------------- /mojito/models/core/net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deezer/sigir23-mojito/HEAD/mojito/models/core/net.py -------------------------------------------------------------------------------- /mojito/models/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deezer/sigir23-mojito/HEAD/mojito/models/model.py -------------------------------------------------------------------------------- /mojito/models/net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deezer/sigir23-mojito/HEAD/mojito/models/net.py -------------------------------------------------------------------------------- /mojito/train/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /mojito/train/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deezer/sigir23-mojito/HEAD/mojito/train/trainer.py -------------------------------------------------------------------------------- /mojito/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deezer/sigir23-mojito/HEAD/mojito/utils/__init__.py -------------------------------------------------------------------------------- /mojito/utils/params.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deezer/sigir23-mojito/HEAD/mojito/utils/params.py -------------------------------------------------------------------------------- /mojito/utils/tempo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deezer/sigir23-mojito/HEAD/mojito/utils/tempo.py -------------------------------------------------------------------------------- /script/run_mojito.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deezer/sigir23-mojito/HEAD/script/run_mojito.sh --------------------------------------------------------------------------------