├── .gitignore ├── LICENSE ├── README.md ├── codabench ├── .env ├── README.md └── codabench_docker.sh ├── examples ├── baseline │ └── ebnerd_feat_baselines.py ├── beyond_accuracy │ └── make_beyond_accuracy.ipynb ├── datasets │ ├── ebnerd_descriptive_analysis.ipynb │ ├── ebnerd_overview.ipynb │ └── plot │ │ ├── article_read_time.png │ │ ├── body_len.png │ │ ├── category_distribution.png │ │ ├── category_distribution_ba.png │ │ ├── front_article_page.png │ │ ├── front_read_time.png │ │ ├── inview_len.png │ │ ├── subtitle_len.png │ │ └── title_len.png ├── quick_start │ ├── lstur_dummy.py │ ├── make_embedding_artifacts.ipynb │ ├── naml_dummy.py │ ├── npa_dummy.py │ ├── nrms_docvec_dummy.py │ ├── nrms_dummy.py │ ├── nrms_ebnerd.ipynb │ └── nrms_ebnerd.py └── reproducibility_scripts │ ├── args_nrms.py │ ├── args_nrms_docvec.py │ ├── ebnerd_nrms.py │ ├── ebnerd_nrms_doc_hist.py │ └── ebnerd_nrms_docvec.py ├── pyproject.toml ├── src ├── __init__.py └── ebrec │ ├── evaluation │ ├── __init__.py │ ├── _ba_test.py │ ├── beyond_accuracy.py │ ├── metrics │ │ ├── __init__.py │ │ ├── _beyond_accuracy.py │ │ ├── _classification.py │ │ ├── _ranking.py │ │ └── _sklearn.py │ ├── metrics_protocols.py │ ├── protocols.py │ └── utils.py │ ├── models │ ├── fastformer │ │ ├── __init__.py │ │ ├── dataloader.py │ │ ├── fastformer.py │ │ └── fastformer_wu.py │ └── newsrec │ │ ├── __init__.py │ │ ├── base_model.py │ │ ├── dataloader.py │ │ ├── layers.py │ │ ├── lstur.py │ │ ├── model_config.py │ │ ├── naml.py │ │ ├── npa.py │ │ ├── nrms.py │ │ ├── nrms_docvec.py │ │ └── utils.py │ └── utils │ ├── __init__.py │ ├── _articles.py │ ├── _articles_behaviors.py │ ├── _behaviors.py │ ├── _constants.py │ ├── _decay.py │ ├── _descriptive_analysis.py │ ├── _nlp.py │ ├── _polars.py │ ├── _python.py │ └── _torch.py └── test ├── bombing └── bomb_dataloader.py ├── data └── ebnerd │ ├── articles.parquet │ ├── behaviors.parquet │ ├── document_vector.parquet │ └── history.parquet ├── dataloader ├── test_fastformer.py └── test_newsrec.py └── evaluation └── test_beyond_accuracy.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ebanalyse/ebnerd-benchmark/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ebanalyse/ebnerd-benchmark/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ebanalyse/ebnerd-benchmark/HEAD/README.md -------------------------------------------------------------------------------- /codabench/.env: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ebanalyse/ebnerd-benchmark/HEAD/codabench/.env -------------------------------------------------------------------------------- /codabench/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ebanalyse/ebnerd-benchmark/HEAD/codabench/README.md -------------------------------------------------------------------------------- /codabench/codabench_docker.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ebanalyse/ebnerd-benchmark/HEAD/codabench/codabench_docker.sh -------------------------------------------------------------------------------- /examples/baseline/ebnerd_feat_baselines.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ebanalyse/ebnerd-benchmark/HEAD/examples/baseline/ebnerd_feat_baselines.py -------------------------------------------------------------------------------- /examples/beyond_accuracy/make_beyond_accuracy.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ebanalyse/ebnerd-benchmark/HEAD/examples/beyond_accuracy/make_beyond_accuracy.ipynb -------------------------------------------------------------------------------- /examples/datasets/ebnerd_descriptive_analysis.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ebanalyse/ebnerd-benchmark/HEAD/examples/datasets/ebnerd_descriptive_analysis.ipynb -------------------------------------------------------------------------------- /examples/datasets/ebnerd_overview.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ebanalyse/ebnerd-benchmark/HEAD/examples/datasets/ebnerd_overview.ipynb -------------------------------------------------------------------------------- /examples/datasets/plot/article_read_time.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ebanalyse/ebnerd-benchmark/HEAD/examples/datasets/plot/article_read_time.png -------------------------------------------------------------------------------- /examples/datasets/plot/body_len.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ebanalyse/ebnerd-benchmark/HEAD/examples/datasets/plot/body_len.png -------------------------------------------------------------------------------- /examples/datasets/plot/category_distribution.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ebanalyse/ebnerd-benchmark/HEAD/examples/datasets/plot/category_distribution.png -------------------------------------------------------------------------------- /examples/datasets/plot/category_distribution_ba.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ebanalyse/ebnerd-benchmark/HEAD/examples/datasets/plot/category_distribution_ba.png -------------------------------------------------------------------------------- /examples/datasets/plot/front_article_page.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ebanalyse/ebnerd-benchmark/HEAD/examples/datasets/plot/front_article_page.png -------------------------------------------------------------------------------- /examples/datasets/plot/front_read_time.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ebanalyse/ebnerd-benchmark/HEAD/examples/datasets/plot/front_read_time.png -------------------------------------------------------------------------------- /examples/datasets/plot/inview_len.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ebanalyse/ebnerd-benchmark/HEAD/examples/datasets/plot/inview_len.png -------------------------------------------------------------------------------- /examples/datasets/plot/subtitle_len.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ebanalyse/ebnerd-benchmark/HEAD/examples/datasets/plot/subtitle_len.png -------------------------------------------------------------------------------- /examples/datasets/plot/title_len.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ebanalyse/ebnerd-benchmark/HEAD/examples/datasets/plot/title_len.png -------------------------------------------------------------------------------- /examples/quick_start/lstur_dummy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ebanalyse/ebnerd-benchmark/HEAD/examples/quick_start/lstur_dummy.py -------------------------------------------------------------------------------- /examples/quick_start/make_embedding_artifacts.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ebanalyse/ebnerd-benchmark/HEAD/examples/quick_start/make_embedding_artifacts.ipynb -------------------------------------------------------------------------------- /examples/quick_start/naml_dummy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ebanalyse/ebnerd-benchmark/HEAD/examples/quick_start/naml_dummy.py -------------------------------------------------------------------------------- /examples/quick_start/npa_dummy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ebanalyse/ebnerd-benchmark/HEAD/examples/quick_start/npa_dummy.py -------------------------------------------------------------------------------- /examples/quick_start/nrms_docvec_dummy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ebanalyse/ebnerd-benchmark/HEAD/examples/quick_start/nrms_docvec_dummy.py -------------------------------------------------------------------------------- /examples/quick_start/nrms_dummy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ebanalyse/ebnerd-benchmark/HEAD/examples/quick_start/nrms_dummy.py -------------------------------------------------------------------------------- /examples/quick_start/nrms_ebnerd.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ebanalyse/ebnerd-benchmark/HEAD/examples/quick_start/nrms_ebnerd.ipynb -------------------------------------------------------------------------------- /examples/quick_start/nrms_ebnerd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ebanalyse/ebnerd-benchmark/HEAD/examples/quick_start/nrms_ebnerd.py -------------------------------------------------------------------------------- /examples/reproducibility_scripts/args_nrms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ebanalyse/ebnerd-benchmark/HEAD/examples/reproducibility_scripts/args_nrms.py -------------------------------------------------------------------------------- /examples/reproducibility_scripts/args_nrms_docvec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ebanalyse/ebnerd-benchmark/HEAD/examples/reproducibility_scripts/args_nrms_docvec.py -------------------------------------------------------------------------------- /examples/reproducibility_scripts/ebnerd_nrms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ebanalyse/ebnerd-benchmark/HEAD/examples/reproducibility_scripts/ebnerd_nrms.py -------------------------------------------------------------------------------- /examples/reproducibility_scripts/ebnerd_nrms_doc_hist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ebanalyse/ebnerd-benchmark/HEAD/examples/reproducibility_scripts/ebnerd_nrms_doc_hist.py -------------------------------------------------------------------------------- /examples/reproducibility_scripts/ebnerd_nrms_docvec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ebanalyse/ebnerd-benchmark/HEAD/examples/reproducibility_scripts/ebnerd_nrms_docvec.py -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ebanalyse/ebnerd-benchmark/HEAD/pyproject.toml -------------------------------------------------------------------------------- /src/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/ebrec/evaluation/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ebanalyse/ebnerd-benchmark/HEAD/src/ebrec/evaluation/__init__.py -------------------------------------------------------------------------------- /src/ebrec/evaluation/_ba_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ebanalyse/ebnerd-benchmark/HEAD/src/ebrec/evaluation/_ba_test.py -------------------------------------------------------------------------------- /src/ebrec/evaluation/beyond_accuracy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ebanalyse/ebnerd-benchmark/HEAD/src/ebrec/evaluation/beyond_accuracy.py -------------------------------------------------------------------------------- /src/ebrec/evaluation/metrics/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ebanalyse/ebnerd-benchmark/HEAD/src/ebrec/evaluation/metrics/__init__.py -------------------------------------------------------------------------------- /src/ebrec/evaluation/metrics/_beyond_accuracy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ebanalyse/ebnerd-benchmark/HEAD/src/ebrec/evaluation/metrics/_beyond_accuracy.py -------------------------------------------------------------------------------- /src/ebrec/evaluation/metrics/_classification.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ebanalyse/ebnerd-benchmark/HEAD/src/ebrec/evaluation/metrics/_classification.py -------------------------------------------------------------------------------- /src/ebrec/evaluation/metrics/_ranking.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ebanalyse/ebnerd-benchmark/HEAD/src/ebrec/evaluation/metrics/_ranking.py -------------------------------------------------------------------------------- /src/ebrec/evaluation/metrics/_sklearn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ebanalyse/ebnerd-benchmark/HEAD/src/ebrec/evaluation/metrics/_sklearn.py -------------------------------------------------------------------------------- /src/ebrec/evaluation/metrics_protocols.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ebanalyse/ebnerd-benchmark/HEAD/src/ebrec/evaluation/metrics_protocols.py -------------------------------------------------------------------------------- /src/ebrec/evaluation/protocols.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ebanalyse/ebnerd-benchmark/HEAD/src/ebrec/evaluation/protocols.py -------------------------------------------------------------------------------- /src/ebrec/evaluation/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ebanalyse/ebnerd-benchmark/HEAD/src/ebrec/evaluation/utils.py -------------------------------------------------------------------------------- /src/ebrec/models/fastformer/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/ebrec/models/fastformer/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ebanalyse/ebnerd-benchmark/HEAD/src/ebrec/models/fastformer/dataloader.py -------------------------------------------------------------------------------- /src/ebrec/models/fastformer/fastformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ebanalyse/ebnerd-benchmark/HEAD/src/ebrec/models/fastformer/fastformer.py -------------------------------------------------------------------------------- /src/ebrec/models/fastformer/fastformer_wu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ebanalyse/ebnerd-benchmark/HEAD/src/ebrec/models/fastformer/fastformer_wu.py -------------------------------------------------------------------------------- /src/ebrec/models/newsrec/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ebanalyse/ebnerd-benchmark/HEAD/src/ebrec/models/newsrec/__init__.py -------------------------------------------------------------------------------- /src/ebrec/models/newsrec/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ebanalyse/ebnerd-benchmark/HEAD/src/ebrec/models/newsrec/base_model.py -------------------------------------------------------------------------------- /src/ebrec/models/newsrec/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ebanalyse/ebnerd-benchmark/HEAD/src/ebrec/models/newsrec/dataloader.py -------------------------------------------------------------------------------- /src/ebrec/models/newsrec/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ebanalyse/ebnerd-benchmark/HEAD/src/ebrec/models/newsrec/layers.py -------------------------------------------------------------------------------- /src/ebrec/models/newsrec/lstur.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ebanalyse/ebnerd-benchmark/HEAD/src/ebrec/models/newsrec/lstur.py -------------------------------------------------------------------------------- /src/ebrec/models/newsrec/model_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ebanalyse/ebnerd-benchmark/HEAD/src/ebrec/models/newsrec/model_config.py -------------------------------------------------------------------------------- /src/ebrec/models/newsrec/naml.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ebanalyse/ebnerd-benchmark/HEAD/src/ebrec/models/newsrec/naml.py -------------------------------------------------------------------------------- /src/ebrec/models/newsrec/npa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ebanalyse/ebnerd-benchmark/HEAD/src/ebrec/models/newsrec/npa.py -------------------------------------------------------------------------------- /src/ebrec/models/newsrec/nrms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ebanalyse/ebnerd-benchmark/HEAD/src/ebrec/models/newsrec/nrms.py -------------------------------------------------------------------------------- /src/ebrec/models/newsrec/nrms_docvec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ebanalyse/ebnerd-benchmark/HEAD/src/ebrec/models/newsrec/nrms_docvec.py -------------------------------------------------------------------------------- /src/ebrec/models/newsrec/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ebanalyse/ebnerd-benchmark/HEAD/src/ebrec/models/newsrec/utils.py -------------------------------------------------------------------------------- /src/ebrec/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/ebrec/utils/_articles.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ebanalyse/ebnerd-benchmark/HEAD/src/ebrec/utils/_articles.py -------------------------------------------------------------------------------- /src/ebrec/utils/_articles_behaviors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ebanalyse/ebnerd-benchmark/HEAD/src/ebrec/utils/_articles_behaviors.py -------------------------------------------------------------------------------- /src/ebrec/utils/_behaviors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ebanalyse/ebnerd-benchmark/HEAD/src/ebrec/utils/_behaviors.py -------------------------------------------------------------------------------- /src/ebrec/utils/_constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ebanalyse/ebnerd-benchmark/HEAD/src/ebrec/utils/_constants.py -------------------------------------------------------------------------------- /src/ebrec/utils/_decay.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ebanalyse/ebnerd-benchmark/HEAD/src/ebrec/utils/_decay.py -------------------------------------------------------------------------------- /src/ebrec/utils/_descriptive_analysis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ebanalyse/ebnerd-benchmark/HEAD/src/ebrec/utils/_descriptive_analysis.py -------------------------------------------------------------------------------- /src/ebrec/utils/_nlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ebanalyse/ebnerd-benchmark/HEAD/src/ebrec/utils/_nlp.py -------------------------------------------------------------------------------- /src/ebrec/utils/_polars.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ebanalyse/ebnerd-benchmark/HEAD/src/ebrec/utils/_polars.py -------------------------------------------------------------------------------- /src/ebrec/utils/_python.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ebanalyse/ebnerd-benchmark/HEAD/src/ebrec/utils/_python.py -------------------------------------------------------------------------------- /src/ebrec/utils/_torch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ebanalyse/ebnerd-benchmark/HEAD/src/ebrec/utils/_torch.py -------------------------------------------------------------------------------- /test/bombing/bomb_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ebanalyse/ebnerd-benchmark/HEAD/test/bombing/bomb_dataloader.py -------------------------------------------------------------------------------- /test/data/ebnerd/articles.parquet: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ebanalyse/ebnerd-benchmark/HEAD/test/data/ebnerd/articles.parquet -------------------------------------------------------------------------------- /test/data/ebnerd/behaviors.parquet: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ebanalyse/ebnerd-benchmark/HEAD/test/data/ebnerd/behaviors.parquet -------------------------------------------------------------------------------- /test/data/ebnerd/document_vector.parquet: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ebanalyse/ebnerd-benchmark/HEAD/test/data/ebnerd/document_vector.parquet -------------------------------------------------------------------------------- /test/data/ebnerd/history.parquet: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ebanalyse/ebnerd-benchmark/HEAD/test/data/ebnerd/history.parquet -------------------------------------------------------------------------------- /test/dataloader/test_fastformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ebanalyse/ebnerd-benchmark/HEAD/test/dataloader/test_fastformer.py -------------------------------------------------------------------------------- /test/dataloader/test_newsrec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ebanalyse/ebnerd-benchmark/HEAD/test/dataloader/test_newsrec.py -------------------------------------------------------------------------------- /test/evaluation/test_beyond_accuracy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ebanalyse/ebnerd-benchmark/HEAD/test/evaluation/test_beyond_accuracy.py --------------------------------------------------------------------------------