├── LICENSE ├── README.md ├── contrib ├── README.md └── sarplus │ ├── DEVELOPMENT.md │ ├── README.md │ ├── azure-pipelines.yml │ ├── python │ ├── pysarplus │ │ ├── SARModel.py │ │ ├── SARPlus.py │ │ └── __init__.py │ ├── setup.py │ ├── src │ │ └── pysarplus.cpp │ └── tests │ │ ├── sample-input.txt │ │ └── test_pyspark_sar.py │ └── scala │ ├── build.sbt │ ├── project │ ├── build.properties │ └── plugins.sbt │ ├── python │ ├── pysarplus_dummy │ │ └── __init__.py │ └── setup.py │ └── src │ ├── main │ └── scala │ │ └── com │ │ └── microsoft │ │ └── sarplus │ │ ├── DefaultSource.scala │ │ ├── SARCacheOutputWriter.scala │ │ └── SARCacheOutputWriterFactory.scala │ └── test │ └── scala │ └── com │ └── microsoft │ └── sarplus │ └── SARCacheOutputWriterSpec.scala ├── data └── data.md ├── docs ├── Makefile ├── README.md └── source │ ├── azureml.rst │ ├── common.rst │ ├── conf.py │ ├── dataset.rst │ ├── evaluation.rst │ ├── index.rst │ ├── recommender.rst │ └── tuning.rst ├── examples ├── 00_quick_start │ ├── README.md │ ├── als_movielens.ipynb │ ├── dkn_MIND.ipynb │ ├── fastai_movielens.ipynb │ ├── geoimc_movielens.ipynb │ ├── lightgbm_tinycriteo.ipynb │ ├── lstur_MIND.ipynb │ ├── naml_MIND.ipynb │ ├── ncf_movielens.ipynb │ ├── npa_MIND.ipynb │ ├── nrms_MIND.ipynb │ ├── rbm_movielens.ipynb │ ├── rlrmc_movielens.ipynb │ ├── sar_movielens.ipynb │ ├── sar_movielens_with_azureml.ipynb │ ├── sar_movieratings_with_azureml_designer.ipynb │ ├── sequential.py │ ├── sequential_recsys_amazondataset.ipynb │ ├── tfidf_covid.ipynb │ ├── wide_deep_movielens.ipynb │ └── xdeepfm_criteo.ipynb ├── 01_prepare_data │ ├── README.md │ ├── data_split.ipynb │ ├── data_transform.ipynb │ └── wikidata_knowledge_graph.ipynb ├── 02_model_collaborative_filtering │ ├── README.md │ ├── als_deep_dive.ipynb │ ├── baseline_deep_dive.ipynb │ ├── cornac_bpr_deep_dive.ipynb │ ├── lightgcn_deep_dive.ipynb │ ├── rbm_deep_dive.ipynb │ ├── sar_deep_dive.ipynb │ └── surprise_svd_deep_dive.ipynb ├── 02_model_content_based_filtering │ ├── README.md │ ├── dkn_deep_dive.ipynb │ ├── mmlspark_lightgbm_criteo.ipynb │ └── vowpal_wabbit_deep_dive.ipynb ├── 02_model_hybrid │ ├── README.md │ ├── fm_deep_dive.ipynb │ ├── lightfm_deep_dive.ipynb │ └── ncf_deep_dive.ipynb ├── 03_evaluate │ ├── README.md │ └── evaluation.ipynb ├── 04_model_select_and_optimize │ ├── README.md │ ├── azureml_hyperdrive_surprise_svd.ipynb │ ├── azureml_hyperdrive_wide_and_deep.ipynb │ ├── nni_ncf.ipynb │ ├── nni_surprise_svd.ipynb │ └── tuning_spark_als.ipynb ├── 05_operationalize │ ├── README.md │ ├── aks_locust_load_test.ipynb │ ├── als_movie_o16n.ipynb │ └── lightgbm_criteo_o16n.ipynb ├── 06_benchmarks │ ├── README.md │ ├── benchmark_utils.py │ └── movielens.ipynb ├── README.md ├── run_notebook_on_azureml.ipynb └── template.ipynb ├── reco_utils ├── .DS_Store ├── README.md ├── __init__.py ├── azureml │ ├── __init__.py │ ├── aks_utils.py │ ├── azureml_designer_modules │ │ ├── entries │ │ │ ├── map_entry.py │ │ │ ├── ndcg_entry.py │ │ │ ├── precision_at_k_entry.py │ │ │ ├── recall_at_k_entry.py │ │ │ ├── score_sar_entry.py │ │ │ ├── stratified_splitter_entry.py │ │ │ └── train_sar_entry.py │ │ └── module_specs │ │ │ ├── map.yaml │ │ │ ├── ndcg.yaml │ │ │ ├── precision_at_k.yaml │ │ │ ├── recall_at_k.yaml │ │ │ ├── sar_conda.yaml │ │ │ ├── sar_score.yaml │ │ │ ├── sar_train.yaml │ │ │ └── stratified_splitter.yaml │ ├── azureml_utils.py │ ├── svd_training.py │ └── wide_deep_training.py ├── common │ ├── __init__.py │ ├── constants.py │ ├── general_utils.py │ ├── gpu_utils.py │ ├── notebook_memory_management.py │ ├── notebook_utils.py │ ├── plot.py │ ├── python_utils.py │ ├── spark_utils.py │ ├── tf_utils.py │ ├── timer.py │ └── visdom_utils.py ├── dataset │ ├── __init__.py │ ├── amazon_reviews.py │ ├── blob_utils.py │ ├── cosmos_cli.py │ ├── covid_utils.py │ ├── criteo.py │ ├── download_utils.py │ ├── mind.py │ ├── movielens.py │ ├── pandas_df_utils.py │ ├── python_splitters.py │ ├── sequential_reviews.py │ ├── spark_splitters.py │ ├── sparse.py │ ├── split_utils.py │ └── wikidata.py ├── evaluation │ ├── __init__.py │ ├── python_evaluation.py │ └── spark_evaluation.py ├── recommender │ ├── .DS_Store │ ├── __init__.py │ ├── cornac │ │ ├── __init__.py │ │ └── cornac_utils.py │ ├── deeprec │ │ ├── .DS_Store │ │ ├── DataModel │ │ │ └── ImplicitCF.py │ │ ├── __init__.py │ │ ├── config │ │ │ ├── asvd.yaml │ │ │ ├── caser.yaml │ │ │ ├── dance.yaml │ │ │ ├── dien.yaml │ │ │ ├── din.yaml │ │ │ ├── gcn.yaml │ │ │ ├── gru4rec.yaml │ │ │ ├── lightgcn.yaml │ │ │ ├── nextitnet.yaml │ │ │ ├── sasli_rec.yaml │ │ │ └── sli_rec.yaml │ │ ├── deeprec_utils.py │ │ ├── io │ │ │ ├── __init__.py │ │ │ ├── dkn_iterator.py │ │ │ ├── iterator.py │ │ │ ├── nextitnet_iterator.py │ │ │ └── sequential_iterator.py │ │ └── models │ │ │ ├── .DS_Store │ │ │ ├── __init__.py │ │ │ ├── base_model.py │ │ │ ├── dkn.py │ │ │ ├── graphrec │ │ │ └── lightgcn.py │ │ │ ├── sequential │ │ │ ├── .DS_Store │ │ │ ├── asvd.py │ │ │ ├── gru4rec.py │ │ │ ├── nextitnet.py │ │ │ ├── rnn_cell_implement.py │ │ │ ├── rnn_dien.py │ │ │ ├── sasli_rec.py │ │ │ ├── sequential_base_model.py │ │ │ ├── sli_rec.py │ │ │ └── surge.py │ │ │ └── xDeepFM.py │ ├── fastai │ │ ├── __init__.py │ │ └── fastai_utils.py │ ├── geoimc │ │ ├── __init__.py │ │ ├── geoimc_algorithm.py │ │ ├── geoimc_data.py │ │ ├── geoimc_predict.py │ │ └── geoimc_utils.py │ ├── lightfm │ │ ├── __init__.py │ │ └── lightfm_utils.py │ ├── lightgbm │ │ ├── __init__.py │ │ └── lightgbm_utils.py │ ├── ncf │ │ ├── __init__.py │ │ ├── dataset.py │ │ └── ncf_singlenode.py │ ├── newsrec │ │ ├── __init__.py │ │ ├── io │ │ │ ├── __init__.py │ │ │ ├── mind_all_iterator.py │ │ │ └── mind_iterator.py │ │ ├── models │ │ │ ├── __init__.py │ │ │ ├── base_model.py │ │ │ ├── layers.py │ │ │ ├── lstur.py │ │ │ ├── naml.py │ │ │ ├── npa.py │ │ │ └── nrms.py │ │ └── newsrec_utils.py │ ├── rbm │ │ ├── __init__.py │ │ └── rbm.py │ ├── rlrmc │ │ ├── RLRMCalgorithm.py │ │ ├── RLRMCdataset.py │ │ ├── __init__.py │ │ └── conjugate_gradient_ms.py │ ├── sar │ │ ├── __init__.py │ │ └── sar_singlenode.py │ ├── surprise │ │ ├── __init__.py │ │ └── surprise_utils.py │ ├── tfidf │ │ ├── __init__.py │ │ └── tfidf_utils.py │ ├── vowpal_wabbit │ │ ├── __init__.py │ │ └── vw.py │ └── wide_deep │ │ ├── __init__.py │ │ └── wide_deep_utils.py └── tuning │ ├── __init__.py │ ├── nni │ ├── ncf_training.py │ ├── ncf_utils.py │ ├── nni_utils.py │ └── svd_training.py │ └── parameter_sweep.py ├── scenarios ├── README.md ├── ads │ └── README.md ├── entertainment │ └── README.md ├── food_and_restaurants │ └── README.md ├── news │ └── README.md ├── retail │ └── README.md └── travel │ └── README.md ├── setup.py ├── tests ├── README.md ├── __init__.py ├── ci │ ├── README.md │ ├── azure_artifact_feed.yaml │ ├── azure_pipeline_test │ │ ├── dsvm_nightly_linux_cpu.yml │ │ ├── dsvm_nightly_linux_gpu.yml │ │ ├── dsvm_nightly_linux_pyspark.yml │ │ ├── dsvm_nightly_win_cpu.yml │ │ ├── dsvm_nightly_win_gpu.yml │ │ ├── dsvm_nightly_win_pyspark.yml │ │ ├── dsvm_notebook_linux_cpu.yml │ │ ├── dsvm_notebook_linux_gpu.yml │ │ ├── dsvm_notebook_linux_pyspark.yml │ │ ├── dsvm_notebook_win_cpu.yml │ │ ├── dsvm_notebook_win_gpu.yml │ │ ├── dsvm_notebook_win_pyspark.yml │ │ ├── dsvm_unit_linux_cpu.yml │ │ ├── dsvm_unit_linux_gpu.yml │ │ ├── dsvm_unit_linux_pyspark.yml │ │ ├── dsvm_unit_win_cpu.yml │ │ ├── dsvm_unit_win_gpu.yml │ │ └── dsvm_unit_win_pyspark.yml │ ├── component_governance.yaml │ ├── cpu_unit_tests.yml │ ├── env-setup.yml │ ├── gpu_unit_test.yml │ ├── nightly_cpu.yml │ ├── nightly_gpu.yml │ ├── notebooks_gpu_unit_tests.yml │ ├── notebooks_unit_tests.yml │ ├── run_pytest.py │ └── submit_azureml_pytest.py ├── conftest.py ├── integration │ ├── test_covid.py │ ├── test_criteo.py │ ├── test_mind.py │ ├── test_movielens.py │ ├── test_notebooks_gpu.py │ ├── test_notebooks_pyspark.py │ └── test_notebooks_python.py ├── ncf_common.py ├── notebooks_common.py ├── rbm_common.py ├── sar_common.py ├── smoke │ ├── test_criteo.py │ ├── test_deeprec_model.py │ ├── test_mind.py │ ├── test_movielens.py │ ├── test_newsrec_model.py │ ├── test_notebooks_gpu.py │ ├── test_notebooks_pyspark.py │ └── test_notebooks_python.py └── unit │ ├── test_aks_utils.py │ ├── test_cornac_utils.py │ ├── test_covid_utils.py │ ├── test_dataset.py │ ├── test_deeprec_model.py │ ├── test_deeprec_utils.py │ ├── test_general_utils.py │ ├── test_geoimc.py │ ├── test_gpu_utils.py │ ├── test_lightfm_utils.py │ ├── test_ncf_dataset.py │ ├── test_ncf_singlenode.py │ ├── test_newsrec_model.py │ ├── test_newsrec_utils.py │ ├── test_nni_utils.py │ ├── test_notebook_utils.ipynb │ ├── test_notebook_utils.py │ ├── test_notebooks_gpu.py │ ├── test_notebooks_pyspark.py │ ├── test_notebooks_python.py │ ├── test_pandas_df_utils.py │ ├── test_plot.py │ ├── test_python_evaluation.py │ ├── test_python_splitter.py │ ├── test_python_utils.py │ ├── test_rbm.py │ ├── test_sar_singlenode.py │ ├── test_spark_evaluation.py │ ├── test_spark_splitter.py │ ├── test_sparse.py │ ├── test_surprise_utils.py │ ├── test_sweep.py │ ├── test_tf_utils.py │ ├── test_tfidf_utils.py │ ├── test_timer.py │ ├── test_vowpal_wabbit.py │ ├── test_wide_deep_utils.py │ └── test_wikidata.py └── tools ├── __init__.py ├── databricks_install.py ├── docker ├── Dockerfile └── README.md ├── generate_conda_file.py └── generate_requirements_txt.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/README.md -------------------------------------------------------------------------------- /contrib/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/contrib/README.md -------------------------------------------------------------------------------- /contrib/sarplus/DEVELOPMENT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/contrib/sarplus/DEVELOPMENT.md -------------------------------------------------------------------------------- /contrib/sarplus/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/contrib/sarplus/README.md -------------------------------------------------------------------------------- /contrib/sarplus/azure-pipelines.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/contrib/sarplus/azure-pipelines.yml -------------------------------------------------------------------------------- /contrib/sarplus/python/pysarplus/SARModel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/contrib/sarplus/python/pysarplus/SARModel.py -------------------------------------------------------------------------------- /contrib/sarplus/python/pysarplus/SARPlus.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/contrib/sarplus/python/pysarplus/SARPlus.py -------------------------------------------------------------------------------- /contrib/sarplus/python/pysarplus/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/contrib/sarplus/python/pysarplus/__init__.py -------------------------------------------------------------------------------- /contrib/sarplus/python/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/contrib/sarplus/python/setup.py -------------------------------------------------------------------------------- /contrib/sarplus/python/src/pysarplus.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/contrib/sarplus/python/src/pysarplus.cpp -------------------------------------------------------------------------------- /contrib/sarplus/python/tests/sample-input.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/contrib/sarplus/python/tests/sample-input.txt -------------------------------------------------------------------------------- /contrib/sarplus/python/tests/test_pyspark_sar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/contrib/sarplus/python/tests/test_pyspark_sar.py -------------------------------------------------------------------------------- /contrib/sarplus/scala/build.sbt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/contrib/sarplus/scala/build.sbt -------------------------------------------------------------------------------- /contrib/sarplus/scala/project/build.properties: -------------------------------------------------------------------------------- 1 | sbt.version=0.13.17 2 | -------------------------------------------------------------------------------- /contrib/sarplus/scala/project/plugins.sbt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/contrib/sarplus/scala/project/plugins.sbt -------------------------------------------------------------------------------- /contrib/sarplus/scala/python/pysarplus_dummy/__init__.py: -------------------------------------------------------------------------------- 1 | installed = 1 2 | -------------------------------------------------------------------------------- /contrib/sarplus/scala/python/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/contrib/sarplus/scala/python/setup.py -------------------------------------------------------------------------------- /contrib/sarplus/scala/src/main/scala/com/microsoft/sarplus/DefaultSource.scala: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/contrib/sarplus/scala/src/main/scala/com/microsoft/sarplus/DefaultSource.scala -------------------------------------------------------------------------------- /contrib/sarplus/scala/src/main/scala/com/microsoft/sarplus/SARCacheOutputWriter.scala: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/contrib/sarplus/scala/src/main/scala/com/microsoft/sarplus/SARCacheOutputWriter.scala -------------------------------------------------------------------------------- /contrib/sarplus/scala/src/main/scala/com/microsoft/sarplus/SARCacheOutputWriterFactory.scala: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/contrib/sarplus/scala/src/main/scala/com/microsoft/sarplus/SARCacheOutputWriterFactory.scala -------------------------------------------------------------------------------- /contrib/sarplus/scala/src/test/scala/com/microsoft/sarplus/SARCacheOutputWriterSpec.scala: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/contrib/sarplus/scala/src/test/scala/com/microsoft/sarplus/SARCacheOutputWriterSpec.scala -------------------------------------------------------------------------------- /data/data.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/data/data.md -------------------------------------------------------------------------------- /docs/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/docs/Makefile -------------------------------------------------------------------------------- /docs/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/docs/README.md -------------------------------------------------------------------------------- /docs/source/azureml.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/docs/source/azureml.rst -------------------------------------------------------------------------------- /docs/source/common.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/docs/source/common.rst -------------------------------------------------------------------------------- /docs/source/conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/docs/source/conf.py -------------------------------------------------------------------------------- /docs/source/dataset.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/docs/source/dataset.rst -------------------------------------------------------------------------------- /docs/source/evaluation.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/docs/source/evaluation.rst -------------------------------------------------------------------------------- /docs/source/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/docs/source/index.rst -------------------------------------------------------------------------------- /docs/source/recommender.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/docs/source/recommender.rst -------------------------------------------------------------------------------- /docs/source/tuning.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/docs/source/tuning.rst -------------------------------------------------------------------------------- /examples/00_quick_start/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/examples/00_quick_start/README.md -------------------------------------------------------------------------------- /examples/00_quick_start/als_movielens.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/examples/00_quick_start/als_movielens.ipynb -------------------------------------------------------------------------------- /examples/00_quick_start/dkn_MIND.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/examples/00_quick_start/dkn_MIND.ipynb -------------------------------------------------------------------------------- /examples/00_quick_start/fastai_movielens.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/examples/00_quick_start/fastai_movielens.ipynb -------------------------------------------------------------------------------- /examples/00_quick_start/geoimc_movielens.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/examples/00_quick_start/geoimc_movielens.ipynb -------------------------------------------------------------------------------- /examples/00_quick_start/lightgbm_tinycriteo.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/examples/00_quick_start/lightgbm_tinycriteo.ipynb -------------------------------------------------------------------------------- /examples/00_quick_start/lstur_MIND.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/examples/00_quick_start/lstur_MIND.ipynb -------------------------------------------------------------------------------- /examples/00_quick_start/naml_MIND.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/examples/00_quick_start/naml_MIND.ipynb -------------------------------------------------------------------------------- /examples/00_quick_start/ncf_movielens.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/examples/00_quick_start/ncf_movielens.ipynb -------------------------------------------------------------------------------- /examples/00_quick_start/npa_MIND.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/examples/00_quick_start/npa_MIND.ipynb -------------------------------------------------------------------------------- /examples/00_quick_start/nrms_MIND.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/examples/00_quick_start/nrms_MIND.ipynb -------------------------------------------------------------------------------- /examples/00_quick_start/rbm_movielens.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/examples/00_quick_start/rbm_movielens.ipynb -------------------------------------------------------------------------------- /examples/00_quick_start/rlrmc_movielens.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/examples/00_quick_start/rlrmc_movielens.ipynb -------------------------------------------------------------------------------- /examples/00_quick_start/sar_movielens.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/examples/00_quick_start/sar_movielens.ipynb -------------------------------------------------------------------------------- /examples/00_quick_start/sar_movielens_with_azureml.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/examples/00_quick_start/sar_movielens_with_azureml.ipynb -------------------------------------------------------------------------------- /examples/00_quick_start/sar_movieratings_with_azureml_designer.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/examples/00_quick_start/sar_movieratings_with_azureml_designer.ipynb -------------------------------------------------------------------------------- /examples/00_quick_start/sequential.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/examples/00_quick_start/sequential.py -------------------------------------------------------------------------------- /examples/00_quick_start/sequential_recsys_amazondataset.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/examples/00_quick_start/sequential_recsys_amazondataset.ipynb -------------------------------------------------------------------------------- /examples/00_quick_start/tfidf_covid.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/examples/00_quick_start/tfidf_covid.ipynb -------------------------------------------------------------------------------- /examples/00_quick_start/wide_deep_movielens.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/examples/00_quick_start/wide_deep_movielens.ipynb -------------------------------------------------------------------------------- /examples/00_quick_start/xdeepfm_criteo.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/examples/00_quick_start/xdeepfm_criteo.ipynb -------------------------------------------------------------------------------- /examples/01_prepare_data/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/examples/01_prepare_data/README.md -------------------------------------------------------------------------------- /examples/01_prepare_data/data_split.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/examples/01_prepare_data/data_split.ipynb -------------------------------------------------------------------------------- /examples/01_prepare_data/data_transform.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/examples/01_prepare_data/data_transform.ipynb -------------------------------------------------------------------------------- /examples/01_prepare_data/wikidata_knowledge_graph.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/examples/01_prepare_data/wikidata_knowledge_graph.ipynb -------------------------------------------------------------------------------- /examples/02_model_collaborative_filtering/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/examples/02_model_collaborative_filtering/README.md -------------------------------------------------------------------------------- /examples/02_model_collaborative_filtering/als_deep_dive.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/examples/02_model_collaborative_filtering/als_deep_dive.ipynb -------------------------------------------------------------------------------- /examples/02_model_collaborative_filtering/baseline_deep_dive.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/examples/02_model_collaborative_filtering/baseline_deep_dive.ipynb -------------------------------------------------------------------------------- /examples/02_model_collaborative_filtering/cornac_bpr_deep_dive.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/examples/02_model_collaborative_filtering/cornac_bpr_deep_dive.ipynb -------------------------------------------------------------------------------- /examples/02_model_collaborative_filtering/lightgcn_deep_dive.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/examples/02_model_collaborative_filtering/lightgcn_deep_dive.ipynb -------------------------------------------------------------------------------- /examples/02_model_collaborative_filtering/rbm_deep_dive.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/examples/02_model_collaborative_filtering/rbm_deep_dive.ipynb -------------------------------------------------------------------------------- /examples/02_model_collaborative_filtering/sar_deep_dive.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/examples/02_model_collaborative_filtering/sar_deep_dive.ipynb -------------------------------------------------------------------------------- /examples/02_model_collaborative_filtering/surprise_svd_deep_dive.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/examples/02_model_collaborative_filtering/surprise_svd_deep_dive.ipynb -------------------------------------------------------------------------------- /examples/02_model_content_based_filtering/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/examples/02_model_content_based_filtering/README.md -------------------------------------------------------------------------------- /examples/02_model_content_based_filtering/dkn_deep_dive.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/examples/02_model_content_based_filtering/dkn_deep_dive.ipynb -------------------------------------------------------------------------------- /examples/02_model_content_based_filtering/mmlspark_lightgbm_criteo.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/examples/02_model_content_based_filtering/mmlspark_lightgbm_criteo.ipynb -------------------------------------------------------------------------------- /examples/02_model_content_based_filtering/vowpal_wabbit_deep_dive.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/examples/02_model_content_based_filtering/vowpal_wabbit_deep_dive.ipynb -------------------------------------------------------------------------------- /examples/02_model_hybrid/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/examples/02_model_hybrid/README.md -------------------------------------------------------------------------------- /examples/02_model_hybrid/fm_deep_dive.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/examples/02_model_hybrid/fm_deep_dive.ipynb -------------------------------------------------------------------------------- /examples/02_model_hybrid/lightfm_deep_dive.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/examples/02_model_hybrid/lightfm_deep_dive.ipynb -------------------------------------------------------------------------------- /examples/02_model_hybrid/ncf_deep_dive.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/examples/02_model_hybrid/ncf_deep_dive.ipynb -------------------------------------------------------------------------------- /examples/03_evaluate/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/examples/03_evaluate/README.md -------------------------------------------------------------------------------- /examples/03_evaluate/evaluation.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/examples/03_evaluate/evaluation.ipynb -------------------------------------------------------------------------------- /examples/04_model_select_and_optimize/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/examples/04_model_select_and_optimize/README.md -------------------------------------------------------------------------------- /examples/04_model_select_and_optimize/azureml_hyperdrive_surprise_svd.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/examples/04_model_select_and_optimize/azureml_hyperdrive_surprise_svd.ipynb -------------------------------------------------------------------------------- /examples/04_model_select_and_optimize/azureml_hyperdrive_wide_and_deep.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/examples/04_model_select_and_optimize/azureml_hyperdrive_wide_and_deep.ipynb -------------------------------------------------------------------------------- /examples/04_model_select_and_optimize/nni_ncf.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/examples/04_model_select_and_optimize/nni_ncf.ipynb -------------------------------------------------------------------------------- /examples/04_model_select_and_optimize/nni_surprise_svd.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/examples/04_model_select_and_optimize/nni_surprise_svd.ipynb -------------------------------------------------------------------------------- /examples/04_model_select_and_optimize/tuning_spark_als.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/examples/04_model_select_and_optimize/tuning_spark_als.ipynb -------------------------------------------------------------------------------- /examples/05_operationalize/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/examples/05_operationalize/README.md -------------------------------------------------------------------------------- /examples/05_operationalize/aks_locust_load_test.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/examples/05_operationalize/aks_locust_load_test.ipynb -------------------------------------------------------------------------------- /examples/05_operationalize/als_movie_o16n.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/examples/05_operationalize/als_movie_o16n.ipynb -------------------------------------------------------------------------------- /examples/05_operationalize/lightgbm_criteo_o16n.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/examples/05_operationalize/lightgbm_criteo_o16n.ipynb -------------------------------------------------------------------------------- /examples/06_benchmarks/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/examples/06_benchmarks/README.md -------------------------------------------------------------------------------- /examples/06_benchmarks/benchmark_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/examples/06_benchmarks/benchmark_utils.py -------------------------------------------------------------------------------- /examples/06_benchmarks/movielens.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/examples/06_benchmarks/movielens.ipynb -------------------------------------------------------------------------------- /examples/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/examples/README.md -------------------------------------------------------------------------------- /examples/run_notebook_on_azureml.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/examples/run_notebook_on_azureml.ipynb -------------------------------------------------------------------------------- /examples/template.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/examples/template.ipynb -------------------------------------------------------------------------------- /reco_utils/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/.DS_Store -------------------------------------------------------------------------------- /reco_utils/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/README.md -------------------------------------------------------------------------------- /reco_utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/__init__.py -------------------------------------------------------------------------------- /reco_utils/azureml/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /reco_utils/azureml/aks_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/azureml/aks_utils.py -------------------------------------------------------------------------------- /reco_utils/azureml/azureml_designer_modules/entries/map_entry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/azureml/azureml_designer_modules/entries/map_entry.py -------------------------------------------------------------------------------- /reco_utils/azureml/azureml_designer_modules/entries/ndcg_entry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/azureml/azureml_designer_modules/entries/ndcg_entry.py -------------------------------------------------------------------------------- /reco_utils/azureml/azureml_designer_modules/entries/precision_at_k_entry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/azureml/azureml_designer_modules/entries/precision_at_k_entry.py -------------------------------------------------------------------------------- /reco_utils/azureml/azureml_designer_modules/entries/recall_at_k_entry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/azureml/azureml_designer_modules/entries/recall_at_k_entry.py -------------------------------------------------------------------------------- /reco_utils/azureml/azureml_designer_modules/entries/score_sar_entry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/azureml/azureml_designer_modules/entries/score_sar_entry.py -------------------------------------------------------------------------------- /reco_utils/azureml/azureml_designer_modules/entries/stratified_splitter_entry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/azureml/azureml_designer_modules/entries/stratified_splitter_entry.py -------------------------------------------------------------------------------- /reco_utils/azureml/azureml_designer_modules/entries/train_sar_entry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/azureml/azureml_designer_modules/entries/train_sar_entry.py -------------------------------------------------------------------------------- /reco_utils/azureml/azureml_designer_modules/module_specs/map.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/azureml/azureml_designer_modules/module_specs/map.yaml -------------------------------------------------------------------------------- /reco_utils/azureml/azureml_designer_modules/module_specs/ndcg.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/azureml/azureml_designer_modules/module_specs/ndcg.yaml -------------------------------------------------------------------------------- /reco_utils/azureml/azureml_designer_modules/module_specs/precision_at_k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/azureml/azureml_designer_modules/module_specs/precision_at_k.yaml -------------------------------------------------------------------------------- /reco_utils/azureml/azureml_designer_modules/module_specs/recall_at_k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/azureml/azureml_designer_modules/module_specs/recall_at_k.yaml -------------------------------------------------------------------------------- /reco_utils/azureml/azureml_designer_modules/module_specs/sar_conda.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/azureml/azureml_designer_modules/module_specs/sar_conda.yaml -------------------------------------------------------------------------------- /reco_utils/azureml/azureml_designer_modules/module_specs/sar_score.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/azureml/azureml_designer_modules/module_specs/sar_score.yaml -------------------------------------------------------------------------------- /reco_utils/azureml/azureml_designer_modules/module_specs/sar_train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/azureml/azureml_designer_modules/module_specs/sar_train.yaml -------------------------------------------------------------------------------- /reco_utils/azureml/azureml_designer_modules/module_specs/stratified_splitter.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/azureml/azureml_designer_modules/module_specs/stratified_splitter.yaml -------------------------------------------------------------------------------- /reco_utils/azureml/azureml_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/azureml/azureml_utils.py -------------------------------------------------------------------------------- /reco_utils/azureml/svd_training.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/azureml/svd_training.py -------------------------------------------------------------------------------- /reco_utils/azureml/wide_deep_training.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/azureml/wide_deep_training.py -------------------------------------------------------------------------------- /reco_utils/common/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /reco_utils/common/constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/common/constants.py -------------------------------------------------------------------------------- /reco_utils/common/general_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/common/general_utils.py -------------------------------------------------------------------------------- /reco_utils/common/gpu_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/common/gpu_utils.py -------------------------------------------------------------------------------- /reco_utils/common/notebook_memory_management.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/common/notebook_memory_management.py -------------------------------------------------------------------------------- /reco_utils/common/notebook_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/common/notebook_utils.py -------------------------------------------------------------------------------- /reco_utils/common/plot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/common/plot.py -------------------------------------------------------------------------------- /reco_utils/common/python_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/common/python_utils.py -------------------------------------------------------------------------------- /reco_utils/common/spark_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/common/spark_utils.py -------------------------------------------------------------------------------- /reco_utils/common/tf_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/common/tf_utils.py -------------------------------------------------------------------------------- /reco_utils/common/timer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/common/timer.py -------------------------------------------------------------------------------- /reco_utils/common/visdom_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/common/visdom_utils.py -------------------------------------------------------------------------------- /reco_utils/dataset/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /reco_utils/dataset/amazon_reviews.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/dataset/amazon_reviews.py -------------------------------------------------------------------------------- /reco_utils/dataset/blob_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/dataset/blob_utils.py -------------------------------------------------------------------------------- /reco_utils/dataset/cosmos_cli.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/dataset/cosmos_cli.py -------------------------------------------------------------------------------- /reco_utils/dataset/covid_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/dataset/covid_utils.py -------------------------------------------------------------------------------- /reco_utils/dataset/criteo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/dataset/criteo.py -------------------------------------------------------------------------------- /reco_utils/dataset/download_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/dataset/download_utils.py -------------------------------------------------------------------------------- /reco_utils/dataset/mind.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/dataset/mind.py -------------------------------------------------------------------------------- /reco_utils/dataset/movielens.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/dataset/movielens.py -------------------------------------------------------------------------------- /reco_utils/dataset/pandas_df_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/dataset/pandas_df_utils.py -------------------------------------------------------------------------------- /reco_utils/dataset/python_splitters.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/dataset/python_splitters.py -------------------------------------------------------------------------------- /reco_utils/dataset/sequential_reviews.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/dataset/sequential_reviews.py -------------------------------------------------------------------------------- /reco_utils/dataset/spark_splitters.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/dataset/spark_splitters.py -------------------------------------------------------------------------------- /reco_utils/dataset/sparse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/dataset/sparse.py -------------------------------------------------------------------------------- /reco_utils/dataset/split_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/dataset/split_utils.py -------------------------------------------------------------------------------- /reco_utils/dataset/wikidata.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/dataset/wikidata.py -------------------------------------------------------------------------------- /reco_utils/evaluation/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /reco_utils/evaluation/python_evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/evaluation/python_evaluation.py -------------------------------------------------------------------------------- /reco_utils/evaluation/spark_evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/evaluation/spark_evaluation.py -------------------------------------------------------------------------------- /reco_utils/recommender/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/recommender/.DS_Store -------------------------------------------------------------------------------- /reco_utils/recommender/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /reco_utils/recommender/cornac/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /reco_utils/recommender/cornac/cornac_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/recommender/cornac/cornac_utils.py -------------------------------------------------------------------------------- /reco_utils/recommender/deeprec/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/recommender/deeprec/.DS_Store -------------------------------------------------------------------------------- /reco_utils/recommender/deeprec/DataModel/ImplicitCF.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/recommender/deeprec/DataModel/ImplicitCF.py -------------------------------------------------------------------------------- /reco_utils/recommender/deeprec/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /reco_utils/recommender/deeprec/config/asvd.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/recommender/deeprec/config/asvd.yaml -------------------------------------------------------------------------------- /reco_utils/recommender/deeprec/config/caser.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/recommender/deeprec/config/caser.yaml -------------------------------------------------------------------------------- /reco_utils/recommender/deeprec/config/dance.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/recommender/deeprec/config/dance.yaml -------------------------------------------------------------------------------- /reco_utils/recommender/deeprec/config/dien.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/recommender/deeprec/config/dien.yaml -------------------------------------------------------------------------------- /reco_utils/recommender/deeprec/config/din.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/recommender/deeprec/config/din.yaml -------------------------------------------------------------------------------- /reco_utils/recommender/deeprec/config/gcn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/recommender/deeprec/config/gcn.yaml -------------------------------------------------------------------------------- /reco_utils/recommender/deeprec/config/gru4rec.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/recommender/deeprec/config/gru4rec.yaml -------------------------------------------------------------------------------- /reco_utils/recommender/deeprec/config/lightgcn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/recommender/deeprec/config/lightgcn.yaml -------------------------------------------------------------------------------- /reco_utils/recommender/deeprec/config/nextitnet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/recommender/deeprec/config/nextitnet.yaml -------------------------------------------------------------------------------- /reco_utils/recommender/deeprec/config/sasli_rec.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/recommender/deeprec/config/sasli_rec.yaml -------------------------------------------------------------------------------- /reco_utils/recommender/deeprec/config/sli_rec.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/recommender/deeprec/config/sli_rec.yaml -------------------------------------------------------------------------------- /reco_utils/recommender/deeprec/deeprec_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/recommender/deeprec/deeprec_utils.py -------------------------------------------------------------------------------- /reco_utils/recommender/deeprec/io/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /reco_utils/recommender/deeprec/io/dkn_iterator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/recommender/deeprec/io/dkn_iterator.py -------------------------------------------------------------------------------- /reco_utils/recommender/deeprec/io/iterator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/recommender/deeprec/io/iterator.py -------------------------------------------------------------------------------- /reco_utils/recommender/deeprec/io/nextitnet_iterator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/recommender/deeprec/io/nextitnet_iterator.py -------------------------------------------------------------------------------- /reco_utils/recommender/deeprec/io/sequential_iterator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/recommender/deeprec/io/sequential_iterator.py -------------------------------------------------------------------------------- /reco_utils/recommender/deeprec/models/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/recommender/deeprec/models/.DS_Store -------------------------------------------------------------------------------- /reco_utils/recommender/deeprec/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /reco_utils/recommender/deeprec/models/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/recommender/deeprec/models/base_model.py -------------------------------------------------------------------------------- /reco_utils/recommender/deeprec/models/dkn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/recommender/deeprec/models/dkn.py -------------------------------------------------------------------------------- /reco_utils/recommender/deeprec/models/graphrec/lightgcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/recommender/deeprec/models/graphrec/lightgcn.py -------------------------------------------------------------------------------- /reco_utils/recommender/deeprec/models/sequential/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/recommender/deeprec/models/sequential/.DS_Store -------------------------------------------------------------------------------- /reco_utils/recommender/deeprec/models/sequential/asvd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/recommender/deeprec/models/sequential/asvd.py -------------------------------------------------------------------------------- /reco_utils/recommender/deeprec/models/sequential/gru4rec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/recommender/deeprec/models/sequential/gru4rec.py -------------------------------------------------------------------------------- /reco_utils/recommender/deeprec/models/sequential/nextitnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/recommender/deeprec/models/sequential/nextitnet.py -------------------------------------------------------------------------------- /reco_utils/recommender/deeprec/models/sequential/rnn_cell_implement.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/recommender/deeprec/models/sequential/rnn_cell_implement.py -------------------------------------------------------------------------------- /reco_utils/recommender/deeprec/models/sequential/rnn_dien.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/recommender/deeprec/models/sequential/rnn_dien.py -------------------------------------------------------------------------------- /reco_utils/recommender/deeprec/models/sequential/sasli_rec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/recommender/deeprec/models/sequential/sasli_rec.py -------------------------------------------------------------------------------- /reco_utils/recommender/deeprec/models/sequential/sequential_base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/recommender/deeprec/models/sequential/sequential_base_model.py -------------------------------------------------------------------------------- /reco_utils/recommender/deeprec/models/sequential/sli_rec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/recommender/deeprec/models/sequential/sli_rec.py -------------------------------------------------------------------------------- /reco_utils/recommender/deeprec/models/sequential/surge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/recommender/deeprec/models/sequential/surge.py -------------------------------------------------------------------------------- /reco_utils/recommender/deeprec/models/xDeepFM.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/recommender/deeprec/models/xDeepFM.py -------------------------------------------------------------------------------- /reco_utils/recommender/fastai/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /reco_utils/recommender/fastai/fastai_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/recommender/fastai/fastai_utils.py -------------------------------------------------------------------------------- /reco_utils/recommender/geoimc/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /reco_utils/recommender/geoimc/geoimc_algorithm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/recommender/geoimc/geoimc_algorithm.py -------------------------------------------------------------------------------- /reco_utils/recommender/geoimc/geoimc_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/recommender/geoimc/geoimc_data.py -------------------------------------------------------------------------------- /reco_utils/recommender/geoimc/geoimc_predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/recommender/geoimc/geoimc_predict.py -------------------------------------------------------------------------------- /reco_utils/recommender/geoimc/geoimc_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/recommender/geoimc/geoimc_utils.py -------------------------------------------------------------------------------- /reco_utils/recommender/lightfm/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /reco_utils/recommender/lightfm/lightfm_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/recommender/lightfm/lightfm_utils.py -------------------------------------------------------------------------------- /reco_utils/recommender/lightgbm/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /reco_utils/recommender/lightgbm/lightgbm_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/recommender/lightgbm/lightgbm_utils.py -------------------------------------------------------------------------------- /reco_utils/recommender/ncf/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /reco_utils/recommender/ncf/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/recommender/ncf/dataset.py -------------------------------------------------------------------------------- /reco_utils/recommender/ncf/ncf_singlenode.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/recommender/ncf/ncf_singlenode.py -------------------------------------------------------------------------------- /reco_utils/recommender/newsrec/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /reco_utils/recommender/newsrec/io/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /reco_utils/recommender/newsrec/io/mind_all_iterator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/recommender/newsrec/io/mind_all_iterator.py -------------------------------------------------------------------------------- /reco_utils/recommender/newsrec/io/mind_iterator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/recommender/newsrec/io/mind_iterator.py -------------------------------------------------------------------------------- /reco_utils/recommender/newsrec/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /reco_utils/recommender/newsrec/models/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/recommender/newsrec/models/base_model.py -------------------------------------------------------------------------------- /reco_utils/recommender/newsrec/models/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/recommender/newsrec/models/layers.py -------------------------------------------------------------------------------- /reco_utils/recommender/newsrec/models/lstur.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/recommender/newsrec/models/lstur.py -------------------------------------------------------------------------------- /reco_utils/recommender/newsrec/models/naml.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/recommender/newsrec/models/naml.py -------------------------------------------------------------------------------- /reco_utils/recommender/newsrec/models/npa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/recommender/newsrec/models/npa.py -------------------------------------------------------------------------------- /reco_utils/recommender/newsrec/models/nrms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/recommender/newsrec/models/nrms.py -------------------------------------------------------------------------------- /reco_utils/recommender/newsrec/newsrec_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/recommender/newsrec/newsrec_utils.py -------------------------------------------------------------------------------- /reco_utils/recommender/rbm/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /reco_utils/recommender/rbm/rbm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/recommender/rbm/rbm.py -------------------------------------------------------------------------------- /reco_utils/recommender/rlrmc/RLRMCalgorithm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/recommender/rlrmc/RLRMCalgorithm.py -------------------------------------------------------------------------------- /reco_utils/recommender/rlrmc/RLRMCdataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/recommender/rlrmc/RLRMCdataset.py -------------------------------------------------------------------------------- /reco_utils/recommender/rlrmc/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /reco_utils/recommender/rlrmc/conjugate_gradient_ms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/recommender/rlrmc/conjugate_gradient_ms.py -------------------------------------------------------------------------------- /reco_utils/recommender/sar/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/recommender/sar/__init__.py -------------------------------------------------------------------------------- /reco_utils/recommender/sar/sar_singlenode.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/recommender/sar/sar_singlenode.py -------------------------------------------------------------------------------- /reco_utils/recommender/surprise/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /reco_utils/recommender/surprise/surprise_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/recommender/surprise/surprise_utils.py -------------------------------------------------------------------------------- /reco_utils/recommender/tfidf/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /reco_utils/recommender/tfidf/tfidf_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/recommender/tfidf/tfidf_utils.py -------------------------------------------------------------------------------- /reco_utils/recommender/vowpal_wabbit/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /reco_utils/recommender/vowpal_wabbit/vw.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/recommender/vowpal_wabbit/vw.py -------------------------------------------------------------------------------- /reco_utils/recommender/wide_deep/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /reco_utils/recommender/wide_deep/wide_deep_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/recommender/wide_deep/wide_deep_utils.py -------------------------------------------------------------------------------- /reco_utils/tuning/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /reco_utils/tuning/nni/ncf_training.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/tuning/nni/ncf_training.py -------------------------------------------------------------------------------- /reco_utils/tuning/nni/ncf_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/tuning/nni/ncf_utils.py -------------------------------------------------------------------------------- /reco_utils/tuning/nni/nni_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/tuning/nni/nni_utils.py -------------------------------------------------------------------------------- /reco_utils/tuning/nni/svd_training.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/tuning/nni/svd_training.py -------------------------------------------------------------------------------- /reco_utils/tuning/parameter_sweep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/reco_utils/tuning/parameter_sweep.py -------------------------------------------------------------------------------- /scenarios/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/scenarios/README.md -------------------------------------------------------------------------------- /scenarios/ads/README.md: -------------------------------------------------------------------------------- 1 | # Recommendation systems for Advertisement 2 | -------------------------------------------------------------------------------- /scenarios/entertainment/README.md: -------------------------------------------------------------------------------- 1 | # Recommendation systems for Entertainment 2 | -------------------------------------------------------------------------------- /scenarios/food_and_restaurants/README.md: -------------------------------------------------------------------------------- 1 | # Recommendation systems for Food and Restaurants 2 | -------------------------------------------------------------------------------- /scenarios/news/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/scenarios/news/README.md -------------------------------------------------------------------------------- /scenarios/retail/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/scenarios/retail/README.md -------------------------------------------------------------------------------- /scenarios/travel/README.md: -------------------------------------------------------------------------------- 1 | # Recommendation systems for Travel 2 | -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/setup.py -------------------------------------------------------------------------------- /tests/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/README.md -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/ci/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/ci/README.md -------------------------------------------------------------------------------- /tests/ci/azure_artifact_feed.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/ci/azure_artifact_feed.yaml -------------------------------------------------------------------------------- /tests/ci/azure_pipeline_test/dsvm_nightly_linux_cpu.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/ci/azure_pipeline_test/dsvm_nightly_linux_cpu.yml -------------------------------------------------------------------------------- /tests/ci/azure_pipeline_test/dsvm_nightly_linux_gpu.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/ci/azure_pipeline_test/dsvm_nightly_linux_gpu.yml -------------------------------------------------------------------------------- /tests/ci/azure_pipeline_test/dsvm_nightly_linux_pyspark.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/ci/azure_pipeline_test/dsvm_nightly_linux_pyspark.yml -------------------------------------------------------------------------------- /tests/ci/azure_pipeline_test/dsvm_nightly_win_cpu.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/ci/azure_pipeline_test/dsvm_nightly_win_cpu.yml -------------------------------------------------------------------------------- /tests/ci/azure_pipeline_test/dsvm_nightly_win_gpu.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/ci/azure_pipeline_test/dsvm_nightly_win_gpu.yml -------------------------------------------------------------------------------- /tests/ci/azure_pipeline_test/dsvm_nightly_win_pyspark.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/ci/azure_pipeline_test/dsvm_nightly_win_pyspark.yml -------------------------------------------------------------------------------- /tests/ci/azure_pipeline_test/dsvm_notebook_linux_cpu.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/ci/azure_pipeline_test/dsvm_notebook_linux_cpu.yml -------------------------------------------------------------------------------- /tests/ci/azure_pipeline_test/dsvm_notebook_linux_gpu.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/ci/azure_pipeline_test/dsvm_notebook_linux_gpu.yml -------------------------------------------------------------------------------- /tests/ci/azure_pipeline_test/dsvm_notebook_linux_pyspark.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/ci/azure_pipeline_test/dsvm_notebook_linux_pyspark.yml -------------------------------------------------------------------------------- /tests/ci/azure_pipeline_test/dsvm_notebook_win_cpu.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/ci/azure_pipeline_test/dsvm_notebook_win_cpu.yml -------------------------------------------------------------------------------- /tests/ci/azure_pipeline_test/dsvm_notebook_win_gpu.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/ci/azure_pipeline_test/dsvm_notebook_win_gpu.yml -------------------------------------------------------------------------------- /tests/ci/azure_pipeline_test/dsvm_notebook_win_pyspark.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/ci/azure_pipeline_test/dsvm_notebook_win_pyspark.yml -------------------------------------------------------------------------------- /tests/ci/azure_pipeline_test/dsvm_unit_linux_cpu.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/ci/azure_pipeline_test/dsvm_unit_linux_cpu.yml -------------------------------------------------------------------------------- /tests/ci/azure_pipeline_test/dsvm_unit_linux_gpu.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/ci/azure_pipeline_test/dsvm_unit_linux_gpu.yml -------------------------------------------------------------------------------- /tests/ci/azure_pipeline_test/dsvm_unit_linux_pyspark.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/ci/azure_pipeline_test/dsvm_unit_linux_pyspark.yml -------------------------------------------------------------------------------- /tests/ci/azure_pipeline_test/dsvm_unit_win_cpu.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/ci/azure_pipeline_test/dsvm_unit_win_cpu.yml -------------------------------------------------------------------------------- /tests/ci/azure_pipeline_test/dsvm_unit_win_gpu.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/ci/azure_pipeline_test/dsvm_unit_win_gpu.yml -------------------------------------------------------------------------------- /tests/ci/azure_pipeline_test/dsvm_unit_win_pyspark.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/ci/azure_pipeline_test/dsvm_unit_win_pyspark.yml -------------------------------------------------------------------------------- /tests/ci/component_governance.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/ci/component_governance.yaml -------------------------------------------------------------------------------- /tests/ci/cpu_unit_tests.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/ci/cpu_unit_tests.yml -------------------------------------------------------------------------------- /tests/ci/env-setup.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/ci/env-setup.yml -------------------------------------------------------------------------------- /tests/ci/gpu_unit_test.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/ci/gpu_unit_test.yml -------------------------------------------------------------------------------- /tests/ci/nightly_cpu.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/ci/nightly_cpu.yml -------------------------------------------------------------------------------- /tests/ci/nightly_gpu.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/ci/nightly_gpu.yml -------------------------------------------------------------------------------- /tests/ci/notebooks_gpu_unit_tests.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/ci/notebooks_gpu_unit_tests.yml -------------------------------------------------------------------------------- /tests/ci/notebooks_unit_tests.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/ci/notebooks_unit_tests.yml -------------------------------------------------------------------------------- /tests/ci/run_pytest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/ci/run_pytest.py -------------------------------------------------------------------------------- /tests/ci/submit_azureml_pytest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/ci/submit_azureml_pytest.py -------------------------------------------------------------------------------- /tests/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/conftest.py -------------------------------------------------------------------------------- /tests/integration/test_covid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/integration/test_covid.py -------------------------------------------------------------------------------- /tests/integration/test_criteo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/integration/test_criteo.py -------------------------------------------------------------------------------- /tests/integration/test_mind.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/integration/test_mind.py -------------------------------------------------------------------------------- /tests/integration/test_movielens.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/integration/test_movielens.py -------------------------------------------------------------------------------- /tests/integration/test_notebooks_gpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/integration/test_notebooks_gpu.py -------------------------------------------------------------------------------- /tests/integration/test_notebooks_pyspark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/integration/test_notebooks_pyspark.py -------------------------------------------------------------------------------- /tests/integration/test_notebooks_python.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/integration/test_notebooks_python.py -------------------------------------------------------------------------------- /tests/ncf_common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/ncf_common.py -------------------------------------------------------------------------------- /tests/notebooks_common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/notebooks_common.py -------------------------------------------------------------------------------- /tests/rbm_common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/rbm_common.py -------------------------------------------------------------------------------- /tests/sar_common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/sar_common.py -------------------------------------------------------------------------------- /tests/smoke/test_criteo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/smoke/test_criteo.py -------------------------------------------------------------------------------- /tests/smoke/test_deeprec_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/smoke/test_deeprec_model.py -------------------------------------------------------------------------------- /tests/smoke/test_mind.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/smoke/test_mind.py -------------------------------------------------------------------------------- /tests/smoke/test_movielens.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/smoke/test_movielens.py -------------------------------------------------------------------------------- /tests/smoke/test_newsrec_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/smoke/test_newsrec_model.py -------------------------------------------------------------------------------- /tests/smoke/test_notebooks_gpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/smoke/test_notebooks_gpu.py -------------------------------------------------------------------------------- /tests/smoke/test_notebooks_pyspark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/smoke/test_notebooks_pyspark.py -------------------------------------------------------------------------------- /tests/smoke/test_notebooks_python.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/smoke/test_notebooks_python.py -------------------------------------------------------------------------------- /tests/unit/test_aks_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/unit/test_aks_utils.py -------------------------------------------------------------------------------- /tests/unit/test_cornac_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/unit/test_cornac_utils.py -------------------------------------------------------------------------------- /tests/unit/test_covid_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/unit/test_covid_utils.py -------------------------------------------------------------------------------- /tests/unit/test_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/unit/test_dataset.py -------------------------------------------------------------------------------- /tests/unit/test_deeprec_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/unit/test_deeprec_model.py -------------------------------------------------------------------------------- /tests/unit/test_deeprec_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/unit/test_deeprec_utils.py -------------------------------------------------------------------------------- /tests/unit/test_general_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/unit/test_general_utils.py -------------------------------------------------------------------------------- /tests/unit/test_geoimc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/unit/test_geoimc.py -------------------------------------------------------------------------------- /tests/unit/test_gpu_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/unit/test_gpu_utils.py -------------------------------------------------------------------------------- /tests/unit/test_lightfm_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/unit/test_lightfm_utils.py -------------------------------------------------------------------------------- /tests/unit/test_ncf_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/unit/test_ncf_dataset.py -------------------------------------------------------------------------------- /tests/unit/test_ncf_singlenode.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/unit/test_ncf_singlenode.py -------------------------------------------------------------------------------- /tests/unit/test_newsrec_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/unit/test_newsrec_model.py -------------------------------------------------------------------------------- /tests/unit/test_newsrec_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/unit/test_newsrec_utils.py -------------------------------------------------------------------------------- /tests/unit/test_nni_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/unit/test_nni_utils.py -------------------------------------------------------------------------------- /tests/unit/test_notebook_utils.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/unit/test_notebook_utils.ipynb -------------------------------------------------------------------------------- /tests/unit/test_notebook_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/unit/test_notebook_utils.py -------------------------------------------------------------------------------- /tests/unit/test_notebooks_gpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/unit/test_notebooks_gpu.py -------------------------------------------------------------------------------- /tests/unit/test_notebooks_pyspark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/unit/test_notebooks_pyspark.py -------------------------------------------------------------------------------- /tests/unit/test_notebooks_python.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/unit/test_notebooks_python.py -------------------------------------------------------------------------------- /tests/unit/test_pandas_df_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/unit/test_pandas_df_utils.py -------------------------------------------------------------------------------- /tests/unit/test_plot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/unit/test_plot.py -------------------------------------------------------------------------------- /tests/unit/test_python_evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/unit/test_python_evaluation.py -------------------------------------------------------------------------------- /tests/unit/test_python_splitter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/unit/test_python_splitter.py -------------------------------------------------------------------------------- /tests/unit/test_python_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/unit/test_python_utils.py -------------------------------------------------------------------------------- /tests/unit/test_rbm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/unit/test_rbm.py -------------------------------------------------------------------------------- /tests/unit/test_sar_singlenode.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/unit/test_sar_singlenode.py -------------------------------------------------------------------------------- /tests/unit/test_spark_evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/unit/test_spark_evaluation.py -------------------------------------------------------------------------------- /tests/unit/test_spark_splitter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/unit/test_spark_splitter.py -------------------------------------------------------------------------------- /tests/unit/test_sparse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/unit/test_sparse.py -------------------------------------------------------------------------------- /tests/unit/test_surprise_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/unit/test_surprise_utils.py -------------------------------------------------------------------------------- /tests/unit/test_sweep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/unit/test_sweep.py -------------------------------------------------------------------------------- /tests/unit/test_tf_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/unit/test_tf_utils.py -------------------------------------------------------------------------------- /tests/unit/test_tfidf_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/unit/test_tfidf_utils.py -------------------------------------------------------------------------------- /tests/unit/test_timer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/unit/test_timer.py -------------------------------------------------------------------------------- /tests/unit/test_vowpal_wabbit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/unit/test_vowpal_wabbit.py -------------------------------------------------------------------------------- /tests/unit/test_wide_deep_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/unit/test_wide_deep_utils.py -------------------------------------------------------------------------------- /tests/unit/test_wikidata.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tests/unit/test_wikidata.py -------------------------------------------------------------------------------- /tools/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tools/databricks_install.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tools/databricks_install.py -------------------------------------------------------------------------------- /tools/docker/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tools/docker/Dockerfile -------------------------------------------------------------------------------- /tools/docker/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tools/docker/README.md -------------------------------------------------------------------------------- /tools/generate_conda_file.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tools/generate_conda_file.py -------------------------------------------------------------------------------- /tools/generate_requirements_txt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tsinghua-fib-lab/SIGIR21-SURGE/HEAD/tools/generate_requirements_txt.py --------------------------------------------------------------------------------