├── .gitignore ├── LICENSE ├── README.md ├── configs ├── callbacks │ ├── default.yaml │ ├── early_stopping.yaml │ ├── model_checkpoint.yaml │ ├── model_summary.yaml │ ├── none.yaml │ └── rich_progress_bar.yaml ├── datamodule │ └── mind.yaml ├── debug │ ├── default.yaml │ ├── fdr.yaml │ ├── limit.yaml │ ├── overfit.yaml │ └── profiler.yaml ├── eval.yaml ├── experiment │ ├── caum_large_celoss_ef.yaml │ ├── caum_large_celoss_lf.yaml │ ├── caum_large_supconloss_ef.yaml │ ├── caum_large_supconloss_lf.yaml │ ├── caum_small_celoss_ef.yaml │ ├── caum_small_celoss_lf.yaml │ ├── caum_small_supconloss_ef.yaml │ ├── caum_small_supconloss_lf.yaml │ ├── cen_news_rec_large_celoss_ef.yaml │ ├── cen_news_rec_large_celoss_lf.yaml │ ├── cen_news_rec_large_supconloss_ef.yaml │ ├── cen_news_rec_large_supconloss_lf.yaml │ ├── cen_news_rec_small_celoss_ef.yaml │ ├── cen_news_rec_small_celoss_lf.yaml │ ├── cen_news_rec_small_supconloss_ef.yaml │ ├── cen_news_rec_small_supconloss_lf.yaml │ ├── dkn_large_celoss_ef.yaml │ ├── dkn_large_celoss_lf.yaml │ ├── dkn_large_supconloss_ef.yaml │ ├── dkn_large_supconloss_lf.yaml │ ├── dkn_small_celoss_ef.yaml │ ├── dkn_small_celoss_lf.yaml │ ├── dkn_small_supconloss_ef.yaml │ ├── dkn_small_supconloss_lf.yaml │ ├── lstur_large_celoss_ef_con.yaml │ ├── lstur_large_celoss_ef_con_tsc.yaml │ ├── lstur_large_celoss_ef_ini.yaml │ ├── lstur_large_celoss_ef_ini_tsc.yaml │ ├── lstur_large_celoss_lf.yaml │ ├── lstur_large_celoss_lf_tsc.yaml │ ├── lstur_large_supconloss_ef_con.yaml │ ├── lstur_large_supconloss_ef_con_tsc.yaml │ ├── lstur_large_supconloss_ef_ini.yaml │ ├── lstur_large_supconloss_ef_ini_tsc.yaml │ ├── lstur_large_supconloss_lf.yaml │ ├── lstur_large_supconloss_lf_tsc.yaml │ ├── lstur_small_celoss_ef_con.yaml │ ├── lstur_small_celoss_ef_con_tsc.yaml │ ├── lstur_small_celoss_ef_ini.yaml │ ├── lstur_small_celoss_ef_ini_tsc.yaml │ ├── lstur_small_celoss_lf.yaml │ ├── lstur_small_celoss_lf_tsc.yaml │ ├── lstur_small_supconloss_ef_con.yaml │ ├── lstur_small_supconloss_ef_con_tsc.yaml │ ├── lstur_small_supconloss_ef_ini.yaml │ ├── lstur_small_supconloss_ef_ini_tsc.yaml │ ├── lstur_small_supconloss_lf.yaml │ ├── lstur_small_supconloss_lf_tsc.yaml │ ├── mins_large_celoss_ef.yaml │ ├── mins_large_celoss_ef_tasc.yaml │ ├── mins_large_celoss_ef_tsc.yaml │ ├── mins_large_celoss_lf.yaml │ ├── mins_large_celoss_lf_tasc.yaml │ ├── mins_large_celoss_lf_tsc.yaml │ ├── mins_large_supconloss_ef.yaml │ ├── mins_large_supconloss_ef_tasc.yaml │ ├── mins_large_supconloss_ef_tsc.yaml │ ├── mins_large_supconloss_lf.yaml │ ├── mins_large_supconloss_lf_tasc.yaml │ ├── mins_large_supconloss_lf_tsc.yaml │ ├── mins_small_celoss_ef.yaml │ ├── mins_small_celoss_ef_tasc.yaml │ ├── mins_small_celoss_ef_tsc.yaml │ ├── mins_small_celoss_lf.yaml │ ├── mins_small_celoss_lf_tasc.yaml │ ├── mins_small_celoss_lf_tsc.yaml │ ├── mins_small_supconloss_ef.yaml │ ├── mins_small_supconloss_ef_tasc.yaml │ ├── mins_small_supconloss_ef_tsc.yaml │ ├── mins_small_supconloss_lf.yaml │ ├── mins_small_supconloss_lf_tasc.yaml │ ├── mins_small_supconloss_lf_tsc.yaml │ ├── naml_large_celoss_ef.yaml │ ├── naml_large_celoss_ef_tasc.yaml │ ├── naml_large_celoss_ef_tsc.yaml │ ├── naml_large_celoss_lf.yaml │ ├── naml_large_celoss_lf_tasc.yaml │ ├── naml_large_celoss_lf_tsc.yaml │ ├── naml_large_supconloss_ef.yaml │ ├── naml_large_supconloss_ef_tasc.yaml │ ├── naml_large_supconloss_ef_tsc.yaml │ ├── naml_large_supconloss_lf.yaml │ ├── naml_large_supconloss_lf_tasc.yaml │ ├── naml_large_supconloss_lf_tsc.yaml │ ├── naml_small_celoss_ef.yaml │ ├── naml_small_celoss_ef_tasc.yaml │ ├── naml_small_celoss_ef_tsc.yaml │ ├── naml_small_celoss_lf.yaml │ ├── naml_small_celoss_lf_tasc.yaml │ ├── naml_small_celoss_lf_tsc.yaml │ ├── naml_small_supconloss_ef.yaml │ ├── naml_small_supconloss_ef_tasc.yaml │ ├── naml_small_supconloss_ef_tsc.yaml │ ├── naml_small_supconloss_lf.yaml │ ├── naml_small_supconloss_lf_tasc.yaml │ ├── naml_small_supconloss_lf_tsc.yaml │ ├── npa_large_celoss_ef.yaml │ ├── npa_large_celoss_lf.yaml │ ├── npa_large_supconloss_ef.yaml │ ├── npa_large_supconloss_lf.yaml │ ├── npa_small_celoss_ef.yaml │ ├── npa_small_celoss_lf.yaml │ ├── npa_small_supconloss_ef.yaml │ ├── npa_small_supconloss_lf.yaml │ ├── nrms_large_celoss_ef.yaml │ ├── nrms_large_celoss_lf.yaml │ ├── nrms_large_supconloss_ef.yaml │ ├── nrms_large_supconloss_lf.yaml │ ├── nrms_small_celoss_ef.yaml │ ├── nrms_small_celoss_lf.yaml │ ├── nrms_small_supconloss_ef.yaml │ └── nrms_small_supconloss_lf.yaml ├── extras │ └── default.yaml ├── hydra │ └── default.yaml ├── local │ └── .gitkeep ├── logger │ ├── csv.yaml │ ├── many_loggers.yaml │ └── wandb.yaml ├── model │ ├── caum.yaml │ ├── cen_news_rec.yaml │ ├── dkn.yaml │ ├── lstur.yaml │ ├── mins.yaml │ ├── naml.yaml │ ├── npa.yaml │ └── nrms.yaml ├── paths │ └── default.yaml ├── train.yaml └── trainer │ ├── cpu.yaml │ ├── ddp.yaml │ ├── ddp_sim.yaml │ ├── default.yaml │ ├── gpu.yaml │ └── mps.yaml ├── data └── .gitkeep ├── framework.png ├── logs └── .gitkeep ├── requirements.txt ├── scripts └── .gitkeep ├── setup.py └── src ├── __init__.py ├── datamodules ├── __init__.py ├── components │ ├── __init__.py │ ├── download_utils.py │ ├── mind_batch.py │ ├── mind_dataframe.py │ └── mind_dataset.py └── mind_datamodule.py ├── eval.py ├── models ├── __init__.py ├── caum │ ├── caum_module.py │ ├── news_encoder.py │ └── user_encoder.py ├── cen_news_rec │ ├── cen_news_rec_module.py │ ├── news_encoder.py │ └── user_encoder.py ├── components │ ├── __init__.py │ ├── attention.py │ ├── click_predictors.py │ ├── losses.py │ └── projections.py ├── dkn │ ├── dkn_module.py │ ├── news_encoder.py │ └── user_encoder.py ├── lstur │ ├── lstur_module.py │ ├── news_encoder.py │ └── user_encoder.py ├── mins │ ├── mins_module.py │ ├── news_encoder.py │ └── user_encoder.py ├── naml │ ├── naml_module.py │ ├── news_encoder.py │ └── user_encoder.py ├── npa │ ├── news_encoder.py │ ├── npa_module.py │ └── user_encoder.py └── nrms │ ├── news_encoder.py │ ├── nrms_module.py │ └── user_encoder.py ├── train.py └── utils ├── __init__.py ├── pylogger.py ├── rich_utils.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/README.md -------------------------------------------------------------------------------- /configs/callbacks/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/callbacks/default.yaml -------------------------------------------------------------------------------- /configs/callbacks/early_stopping.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/callbacks/early_stopping.yaml -------------------------------------------------------------------------------- /configs/callbacks/model_checkpoint.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/callbacks/model_checkpoint.yaml -------------------------------------------------------------------------------- /configs/callbacks/model_summary.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/callbacks/model_summary.yaml -------------------------------------------------------------------------------- /configs/callbacks/none.yaml: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /configs/callbacks/rich_progress_bar.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/callbacks/rich_progress_bar.yaml -------------------------------------------------------------------------------- /configs/datamodule/mind.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/datamodule/mind.yaml -------------------------------------------------------------------------------- /configs/debug/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/debug/default.yaml -------------------------------------------------------------------------------- /configs/debug/fdr.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/debug/fdr.yaml -------------------------------------------------------------------------------- /configs/debug/limit.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/debug/limit.yaml -------------------------------------------------------------------------------- /configs/debug/overfit.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/debug/overfit.yaml -------------------------------------------------------------------------------- /configs/debug/profiler.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/debug/profiler.yaml -------------------------------------------------------------------------------- /configs/eval.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/eval.yaml -------------------------------------------------------------------------------- /configs/experiment/caum_large_celoss_ef.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/caum_large_celoss_ef.yaml -------------------------------------------------------------------------------- /configs/experiment/caum_large_celoss_lf.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/caum_large_celoss_lf.yaml -------------------------------------------------------------------------------- /configs/experiment/caum_large_supconloss_ef.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/caum_large_supconloss_ef.yaml -------------------------------------------------------------------------------- /configs/experiment/caum_large_supconloss_lf.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/caum_large_supconloss_lf.yaml -------------------------------------------------------------------------------- /configs/experiment/caum_small_celoss_ef.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/caum_small_celoss_ef.yaml -------------------------------------------------------------------------------- /configs/experiment/caum_small_celoss_lf.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/caum_small_celoss_lf.yaml -------------------------------------------------------------------------------- /configs/experiment/caum_small_supconloss_ef.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/caum_small_supconloss_ef.yaml -------------------------------------------------------------------------------- /configs/experiment/caum_small_supconloss_lf.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/caum_small_supconloss_lf.yaml -------------------------------------------------------------------------------- /configs/experiment/cen_news_rec_large_celoss_ef.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/cen_news_rec_large_celoss_ef.yaml -------------------------------------------------------------------------------- /configs/experiment/cen_news_rec_large_celoss_lf.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/cen_news_rec_large_celoss_lf.yaml -------------------------------------------------------------------------------- /configs/experiment/cen_news_rec_large_supconloss_ef.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/cen_news_rec_large_supconloss_ef.yaml -------------------------------------------------------------------------------- /configs/experiment/cen_news_rec_large_supconloss_lf.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/cen_news_rec_large_supconloss_lf.yaml -------------------------------------------------------------------------------- /configs/experiment/cen_news_rec_small_celoss_ef.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/cen_news_rec_small_celoss_ef.yaml -------------------------------------------------------------------------------- /configs/experiment/cen_news_rec_small_celoss_lf.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/cen_news_rec_small_celoss_lf.yaml -------------------------------------------------------------------------------- /configs/experiment/cen_news_rec_small_supconloss_ef.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/cen_news_rec_small_supconloss_ef.yaml -------------------------------------------------------------------------------- /configs/experiment/cen_news_rec_small_supconloss_lf.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/cen_news_rec_small_supconloss_lf.yaml -------------------------------------------------------------------------------- /configs/experiment/dkn_large_celoss_ef.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/dkn_large_celoss_ef.yaml -------------------------------------------------------------------------------- /configs/experiment/dkn_large_celoss_lf.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/dkn_large_celoss_lf.yaml -------------------------------------------------------------------------------- /configs/experiment/dkn_large_supconloss_ef.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/dkn_large_supconloss_ef.yaml -------------------------------------------------------------------------------- /configs/experiment/dkn_large_supconloss_lf.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/dkn_large_supconloss_lf.yaml -------------------------------------------------------------------------------- /configs/experiment/dkn_small_celoss_ef.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/dkn_small_celoss_ef.yaml -------------------------------------------------------------------------------- /configs/experiment/dkn_small_celoss_lf.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/dkn_small_celoss_lf.yaml -------------------------------------------------------------------------------- /configs/experiment/dkn_small_supconloss_ef.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/dkn_small_supconloss_ef.yaml -------------------------------------------------------------------------------- /configs/experiment/dkn_small_supconloss_lf.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/dkn_small_supconloss_lf.yaml -------------------------------------------------------------------------------- /configs/experiment/lstur_large_celoss_ef_con.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/lstur_large_celoss_ef_con.yaml -------------------------------------------------------------------------------- /configs/experiment/lstur_large_celoss_ef_con_tsc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/lstur_large_celoss_ef_con_tsc.yaml -------------------------------------------------------------------------------- /configs/experiment/lstur_large_celoss_ef_ini.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/lstur_large_celoss_ef_ini.yaml -------------------------------------------------------------------------------- /configs/experiment/lstur_large_celoss_ef_ini_tsc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/lstur_large_celoss_ef_ini_tsc.yaml -------------------------------------------------------------------------------- /configs/experiment/lstur_large_celoss_lf.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/lstur_large_celoss_lf.yaml -------------------------------------------------------------------------------- /configs/experiment/lstur_large_celoss_lf_tsc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/lstur_large_celoss_lf_tsc.yaml -------------------------------------------------------------------------------- /configs/experiment/lstur_large_supconloss_ef_con.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/lstur_large_supconloss_ef_con.yaml -------------------------------------------------------------------------------- /configs/experiment/lstur_large_supconloss_ef_con_tsc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/lstur_large_supconloss_ef_con_tsc.yaml -------------------------------------------------------------------------------- /configs/experiment/lstur_large_supconloss_ef_ini.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/lstur_large_supconloss_ef_ini.yaml -------------------------------------------------------------------------------- /configs/experiment/lstur_large_supconloss_ef_ini_tsc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/lstur_large_supconloss_ef_ini_tsc.yaml -------------------------------------------------------------------------------- /configs/experiment/lstur_large_supconloss_lf.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/lstur_large_supconloss_lf.yaml -------------------------------------------------------------------------------- /configs/experiment/lstur_large_supconloss_lf_tsc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/lstur_large_supconloss_lf_tsc.yaml -------------------------------------------------------------------------------- /configs/experiment/lstur_small_celoss_ef_con.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/lstur_small_celoss_ef_con.yaml -------------------------------------------------------------------------------- /configs/experiment/lstur_small_celoss_ef_con_tsc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/lstur_small_celoss_ef_con_tsc.yaml -------------------------------------------------------------------------------- /configs/experiment/lstur_small_celoss_ef_ini.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/lstur_small_celoss_ef_ini.yaml -------------------------------------------------------------------------------- /configs/experiment/lstur_small_celoss_ef_ini_tsc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/lstur_small_celoss_ef_ini_tsc.yaml -------------------------------------------------------------------------------- /configs/experiment/lstur_small_celoss_lf.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/lstur_small_celoss_lf.yaml -------------------------------------------------------------------------------- /configs/experiment/lstur_small_celoss_lf_tsc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/lstur_small_celoss_lf_tsc.yaml -------------------------------------------------------------------------------- /configs/experiment/lstur_small_supconloss_ef_con.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/lstur_small_supconloss_ef_con.yaml -------------------------------------------------------------------------------- /configs/experiment/lstur_small_supconloss_ef_con_tsc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/lstur_small_supconloss_ef_con_tsc.yaml -------------------------------------------------------------------------------- /configs/experiment/lstur_small_supconloss_ef_ini.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/lstur_small_supconloss_ef_ini.yaml -------------------------------------------------------------------------------- /configs/experiment/lstur_small_supconloss_ef_ini_tsc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/lstur_small_supconloss_ef_ini_tsc.yaml -------------------------------------------------------------------------------- /configs/experiment/lstur_small_supconloss_lf.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/lstur_small_supconloss_lf.yaml -------------------------------------------------------------------------------- /configs/experiment/lstur_small_supconloss_lf_tsc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/lstur_small_supconloss_lf_tsc.yaml -------------------------------------------------------------------------------- /configs/experiment/mins_large_celoss_ef.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/mins_large_celoss_ef.yaml -------------------------------------------------------------------------------- /configs/experiment/mins_large_celoss_ef_tasc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/mins_large_celoss_ef_tasc.yaml -------------------------------------------------------------------------------- /configs/experiment/mins_large_celoss_ef_tsc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/mins_large_celoss_ef_tsc.yaml -------------------------------------------------------------------------------- /configs/experiment/mins_large_celoss_lf.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/mins_large_celoss_lf.yaml -------------------------------------------------------------------------------- /configs/experiment/mins_large_celoss_lf_tasc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/mins_large_celoss_lf_tasc.yaml -------------------------------------------------------------------------------- /configs/experiment/mins_large_celoss_lf_tsc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/mins_large_celoss_lf_tsc.yaml -------------------------------------------------------------------------------- /configs/experiment/mins_large_supconloss_ef.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/mins_large_supconloss_ef.yaml -------------------------------------------------------------------------------- /configs/experiment/mins_large_supconloss_ef_tasc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/mins_large_supconloss_ef_tasc.yaml -------------------------------------------------------------------------------- /configs/experiment/mins_large_supconloss_ef_tsc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/mins_large_supconloss_ef_tsc.yaml -------------------------------------------------------------------------------- /configs/experiment/mins_large_supconloss_lf.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/mins_large_supconloss_lf.yaml -------------------------------------------------------------------------------- /configs/experiment/mins_large_supconloss_lf_tasc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/mins_large_supconloss_lf_tasc.yaml -------------------------------------------------------------------------------- /configs/experiment/mins_large_supconloss_lf_tsc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/mins_large_supconloss_lf_tsc.yaml -------------------------------------------------------------------------------- /configs/experiment/mins_small_celoss_ef.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/mins_small_celoss_ef.yaml -------------------------------------------------------------------------------- /configs/experiment/mins_small_celoss_ef_tasc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/mins_small_celoss_ef_tasc.yaml -------------------------------------------------------------------------------- /configs/experiment/mins_small_celoss_ef_tsc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/mins_small_celoss_ef_tsc.yaml -------------------------------------------------------------------------------- /configs/experiment/mins_small_celoss_lf.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/mins_small_celoss_lf.yaml -------------------------------------------------------------------------------- /configs/experiment/mins_small_celoss_lf_tasc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/mins_small_celoss_lf_tasc.yaml -------------------------------------------------------------------------------- /configs/experiment/mins_small_celoss_lf_tsc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/mins_small_celoss_lf_tsc.yaml -------------------------------------------------------------------------------- /configs/experiment/mins_small_supconloss_ef.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/mins_small_supconloss_ef.yaml -------------------------------------------------------------------------------- /configs/experiment/mins_small_supconloss_ef_tasc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/mins_small_supconloss_ef_tasc.yaml -------------------------------------------------------------------------------- /configs/experiment/mins_small_supconloss_ef_tsc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/mins_small_supconloss_ef_tsc.yaml -------------------------------------------------------------------------------- /configs/experiment/mins_small_supconloss_lf.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/mins_small_supconloss_lf.yaml -------------------------------------------------------------------------------- /configs/experiment/mins_small_supconloss_lf_tasc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/mins_small_supconloss_lf_tasc.yaml -------------------------------------------------------------------------------- /configs/experiment/mins_small_supconloss_lf_tsc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/mins_small_supconloss_lf_tsc.yaml -------------------------------------------------------------------------------- /configs/experiment/naml_large_celoss_ef.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/naml_large_celoss_ef.yaml -------------------------------------------------------------------------------- /configs/experiment/naml_large_celoss_ef_tasc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/naml_large_celoss_ef_tasc.yaml -------------------------------------------------------------------------------- /configs/experiment/naml_large_celoss_ef_tsc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/naml_large_celoss_ef_tsc.yaml -------------------------------------------------------------------------------- /configs/experiment/naml_large_celoss_lf.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/naml_large_celoss_lf.yaml -------------------------------------------------------------------------------- /configs/experiment/naml_large_celoss_lf_tasc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/naml_large_celoss_lf_tasc.yaml -------------------------------------------------------------------------------- /configs/experiment/naml_large_celoss_lf_tsc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/naml_large_celoss_lf_tsc.yaml -------------------------------------------------------------------------------- /configs/experiment/naml_large_supconloss_ef.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/naml_large_supconloss_ef.yaml -------------------------------------------------------------------------------- /configs/experiment/naml_large_supconloss_ef_tasc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/naml_large_supconloss_ef_tasc.yaml -------------------------------------------------------------------------------- /configs/experiment/naml_large_supconloss_ef_tsc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/naml_large_supconloss_ef_tsc.yaml -------------------------------------------------------------------------------- /configs/experiment/naml_large_supconloss_lf.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/naml_large_supconloss_lf.yaml -------------------------------------------------------------------------------- /configs/experiment/naml_large_supconloss_lf_tasc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/naml_large_supconloss_lf_tasc.yaml -------------------------------------------------------------------------------- /configs/experiment/naml_large_supconloss_lf_tsc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/naml_large_supconloss_lf_tsc.yaml -------------------------------------------------------------------------------- /configs/experiment/naml_small_celoss_ef.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/naml_small_celoss_ef.yaml -------------------------------------------------------------------------------- /configs/experiment/naml_small_celoss_ef_tasc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/naml_small_celoss_ef_tasc.yaml -------------------------------------------------------------------------------- /configs/experiment/naml_small_celoss_ef_tsc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/naml_small_celoss_ef_tsc.yaml -------------------------------------------------------------------------------- /configs/experiment/naml_small_celoss_lf.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/naml_small_celoss_lf.yaml -------------------------------------------------------------------------------- /configs/experiment/naml_small_celoss_lf_tasc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/naml_small_celoss_lf_tasc.yaml -------------------------------------------------------------------------------- /configs/experiment/naml_small_celoss_lf_tsc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/naml_small_celoss_lf_tsc.yaml -------------------------------------------------------------------------------- /configs/experiment/naml_small_supconloss_ef.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/naml_small_supconloss_ef.yaml -------------------------------------------------------------------------------- /configs/experiment/naml_small_supconloss_ef_tasc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/naml_small_supconloss_ef_tasc.yaml -------------------------------------------------------------------------------- /configs/experiment/naml_small_supconloss_ef_tsc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/naml_small_supconloss_ef_tsc.yaml -------------------------------------------------------------------------------- /configs/experiment/naml_small_supconloss_lf.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/naml_small_supconloss_lf.yaml -------------------------------------------------------------------------------- /configs/experiment/naml_small_supconloss_lf_tasc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/naml_small_supconloss_lf_tasc.yaml -------------------------------------------------------------------------------- /configs/experiment/naml_small_supconloss_lf_tsc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/naml_small_supconloss_lf_tsc.yaml -------------------------------------------------------------------------------- /configs/experiment/npa_large_celoss_ef.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/npa_large_celoss_ef.yaml -------------------------------------------------------------------------------- /configs/experiment/npa_large_celoss_lf.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/npa_large_celoss_lf.yaml -------------------------------------------------------------------------------- /configs/experiment/npa_large_supconloss_ef.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/npa_large_supconloss_ef.yaml -------------------------------------------------------------------------------- /configs/experiment/npa_large_supconloss_lf.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/npa_large_supconloss_lf.yaml -------------------------------------------------------------------------------- /configs/experiment/npa_small_celoss_ef.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/npa_small_celoss_ef.yaml -------------------------------------------------------------------------------- /configs/experiment/npa_small_celoss_lf.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/npa_small_celoss_lf.yaml -------------------------------------------------------------------------------- /configs/experiment/npa_small_supconloss_ef.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/npa_small_supconloss_ef.yaml -------------------------------------------------------------------------------- /configs/experiment/npa_small_supconloss_lf.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/npa_small_supconloss_lf.yaml -------------------------------------------------------------------------------- /configs/experiment/nrms_large_celoss_ef.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/nrms_large_celoss_ef.yaml -------------------------------------------------------------------------------- /configs/experiment/nrms_large_celoss_lf.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/nrms_large_celoss_lf.yaml -------------------------------------------------------------------------------- /configs/experiment/nrms_large_supconloss_ef.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/nrms_large_supconloss_ef.yaml -------------------------------------------------------------------------------- /configs/experiment/nrms_large_supconloss_lf.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/nrms_large_supconloss_lf.yaml -------------------------------------------------------------------------------- /configs/experiment/nrms_small_celoss_ef.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/nrms_small_celoss_ef.yaml -------------------------------------------------------------------------------- /configs/experiment/nrms_small_celoss_lf.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/nrms_small_celoss_lf.yaml -------------------------------------------------------------------------------- /configs/experiment/nrms_small_supconloss_ef.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/nrms_small_supconloss_ef.yaml -------------------------------------------------------------------------------- /configs/experiment/nrms_small_supconloss_lf.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/experiment/nrms_small_supconloss_lf.yaml -------------------------------------------------------------------------------- /configs/extras/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/extras/default.yaml -------------------------------------------------------------------------------- /configs/hydra/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/hydra/default.yaml -------------------------------------------------------------------------------- /configs/local/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /configs/logger/csv.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/logger/csv.yaml -------------------------------------------------------------------------------- /configs/logger/many_loggers.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/logger/many_loggers.yaml -------------------------------------------------------------------------------- /configs/logger/wandb.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/logger/wandb.yaml -------------------------------------------------------------------------------- /configs/model/caum.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/model/caum.yaml -------------------------------------------------------------------------------- /configs/model/cen_news_rec.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/model/cen_news_rec.yaml -------------------------------------------------------------------------------- /configs/model/dkn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/model/dkn.yaml -------------------------------------------------------------------------------- /configs/model/lstur.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/model/lstur.yaml -------------------------------------------------------------------------------- /configs/model/mins.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/model/mins.yaml -------------------------------------------------------------------------------- /configs/model/naml.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/model/naml.yaml -------------------------------------------------------------------------------- /configs/model/npa.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/model/npa.yaml -------------------------------------------------------------------------------- /configs/model/nrms.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/model/nrms.yaml -------------------------------------------------------------------------------- /configs/paths/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/paths/default.yaml -------------------------------------------------------------------------------- /configs/train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/train.yaml -------------------------------------------------------------------------------- /configs/trainer/cpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/trainer/cpu.yaml -------------------------------------------------------------------------------- /configs/trainer/ddp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/trainer/ddp.yaml -------------------------------------------------------------------------------- /configs/trainer/ddp_sim.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/trainer/ddp_sim.yaml -------------------------------------------------------------------------------- /configs/trainer/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/trainer/default.yaml -------------------------------------------------------------------------------- /configs/trainer/gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/trainer/gpu.yaml -------------------------------------------------------------------------------- /configs/trainer/mps.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/configs/trainer/mps.yaml -------------------------------------------------------------------------------- /data/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /framework.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/framework.png -------------------------------------------------------------------------------- /logs/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/setup.py -------------------------------------------------------------------------------- /src/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/datamodules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/datamodules/components/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/datamodules/components/download_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/src/datamodules/components/download_utils.py -------------------------------------------------------------------------------- /src/datamodules/components/mind_batch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/src/datamodules/components/mind_batch.py -------------------------------------------------------------------------------- /src/datamodules/components/mind_dataframe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/src/datamodules/components/mind_dataframe.py -------------------------------------------------------------------------------- /src/datamodules/components/mind_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/src/datamodules/components/mind_dataset.py -------------------------------------------------------------------------------- /src/datamodules/mind_datamodule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/src/datamodules/mind_datamodule.py -------------------------------------------------------------------------------- /src/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/src/eval.py -------------------------------------------------------------------------------- /src/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/models/caum/caum_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/src/models/caum/caum_module.py -------------------------------------------------------------------------------- /src/models/caum/news_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/src/models/caum/news_encoder.py -------------------------------------------------------------------------------- /src/models/caum/user_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/src/models/caum/user_encoder.py -------------------------------------------------------------------------------- /src/models/cen_news_rec/cen_news_rec_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/src/models/cen_news_rec/cen_news_rec_module.py -------------------------------------------------------------------------------- /src/models/cen_news_rec/news_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/src/models/cen_news_rec/news_encoder.py -------------------------------------------------------------------------------- /src/models/cen_news_rec/user_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/src/models/cen_news_rec/user_encoder.py -------------------------------------------------------------------------------- /src/models/components/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/models/components/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/src/models/components/attention.py -------------------------------------------------------------------------------- /src/models/components/click_predictors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/src/models/components/click_predictors.py -------------------------------------------------------------------------------- /src/models/components/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/src/models/components/losses.py -------------------------------------------------------------------------------- /src/models/components/projections.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/src/models/components/projections.py -------------------------------------------------------------------------------- /src/models/dkn/dkn_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/src/models/dkn/dkn_module.py -------------------------------------------------------------------------------- /src/models/dkn/news_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/src/models/dkn/news_encoder.py -------------------------------------------------------------------------------- /src/models/dkn/user_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/src/models/dkn/user_encoder.py -------------------------------------------------------------------------------- /src/models/lstur/lstur_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/src/models/lstur/lstur_module.py -------------------------------------------------------------------------------- /src/models/lstur/news_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/src/models/lstur/news_encoder.py -------------------------------------------------------------------------------- /src/models/lstur/user_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/src/models/lstur/user_encoder.py -------------------------------------------------------------------------------- /src/models/mins/mins_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/src/models/mins/mins_module.py -------------------------------------------------------------------------------- /src/models/mins/news_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/src/models/mins/news_encoder.py -------------------------------------------------------------------------------- /src/models/mins/user_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/src/models/mins/user_encoder.py -------------------------------------------------------------------------------- /src/models/naml/naml_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/src/models/naml/naml_module.py -------------------------------------------------------------------------------- /src/models/naml/news_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/src/models/naml/news_encoder.py -------------------------------------------------------------------------------- /src/models/naml/user_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/src/models/naml/user_encoder.py -------------------------------------------------------------------------------- /src/models/npa/news_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/src/models/npa/news_encoder.py -------------------------------------------------------------------------------- /src/models/npa/npa_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/src/models/npa/npa_module.py -------------------------------------------------------------------------------- /src/models/npa/user_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/src/models/npa/user_encoder.py -------------------------------------------------------------------------------- /src/models/nrms/news_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/src/models/nrms/news_encoder.py -------------------------------------------------------------------------------- /src/models/nrms/nrms_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/src/models/nrms/nrms_module.py -------------------------------------------------------------------------------- /src/models/nrms/user_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/src/models/nrms/user_encoder.py -------------------------------------------------------------------------------- /src/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/src/train.py -------------------------------------------------------------------------------- /src/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/src/utils/__init__.py -------------------------------------------------------------------------------- /src/utils/pylogger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/src/utils/pylogger.py -------------------------------------------------------------------------------- /src/utils/rich_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/src/utils/rich_utils.py -------------------------------------------------------------------------------- /src/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/andreeaiana/simplifying_nnr/HEAD/src/utils/utils.py --------------------------------------------------------------------------------