├── Data_Javier ├── ml-1m.test.negative ├── ml-1m.test.rating └── ml-1m.train.rating ├── Data_Xiangnan ├── ml-1m.test.negative ├── ml-1m.test.rating └── ml-1m.train.rating ├── Data_raw └── ml-1m │ ├── README │ ├── movies.dat │ ├── ratings.dat │ └── users.dat ├── Dataset.py ├── GMF_gluon.py ├── GMF_keras.py ├── GMF_pytorch.py ├── MLP_gluon.py ├── MLP_keras.py ├── MLP_pytorch.py ├── NeuMF_gluon.py ├── NeuMF_keras.py ├── NeuMF_pytorch.py ├── README.md ├── data_comparison.ipynb ├── data_preparation.py ├── docs └── images │ ├── GMF_model.png │ ├── MLP_model.png │ ├── NeuMF_model.png │ ├── metrics_vs_n_emb.png │ └── train_time_vs_n_emb.png ├── evaluate_keras.py ├── models ├── gluon_GMF_bs_1024_lr_0001_n_emb_32.params ├── gluon_MLP_bs_512_reg_00_lr_001_n_emb_128_ll_64_dp_wodp.params ├── keras_GMF_bs_1024_reg_00_lr_0001_n_emb_16.h5 ├── keras_MLP_bs_256_reg_00_lr_0001_n_emb_128_ll_64_dp_wodp.h5 ├── pytorch_GMF_bs_256_lr_0001_n_emb_32.pt ├── pytorch_MLP_bs_256_reg_00_lr_001_n_emb_64_ll_32_dp_wodp.pt └── results_df.csv ├── plot_utils.py ├── results_summary.ipynb ├── run_net.sh └── utils.py /Data_Javier/ml-1m.test.negative: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrzaurin/neural_cf/HEAD/Data_Javier/ml-1m.test.negative -------------------------------------------------------------------------------- /Data_Javier/ml-1m.test.rating: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrzaurin/neural_cf/HEAD/Data_Javier/ml-1m.test.rating -------------------------------------------------------------------------------- /Data_Javier/ml-1m.train.rating: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrzaurin/neural_cf/HEAD/Data_Javier/ml-1m.train.rating -------------------------------------------------------------------------------- /Data_Xiangnan/ml-1m.test.negative: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrzaurin/neural_cf/HEAD/Data_Xiangnan/ml-1m.test.negative -------------------------------------------------------------------------------- /Data_Xiangnan/ml-1m.test.rating: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrzaurin/neural_cf/HEAD/Data_Xiangnan/ml-1m.test.rating -------------------------------------------------------------------------------- /Data_Xiangnan/ml-1m.train.rating: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrzaurin/neural_cf/HEAD/Data_Xiangnan/ml-1m.train.rating -------------------------------------------------------------------------------- /Data_raw/ml-1m/README: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrzaurin/neural_cf/HEAD/Data_raw/ml-1m/README -------------------------------------------------------------------------------- /Data_raw/ml-1m/movies.dat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrzaurin/neural_cf/HEAD/Data_raw/ml-1m/movies.dat -------------------------------------------------------------------------------- /Data_raw/ml-1m/ratings.dat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrzaurin/neural_cf/HEAD/Data_raw/ml-1m/ratings.dat -------------------------------------------------------------------------------- /Data_raw/ml-1m/users.dat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrzaurin/neural_cf/HEAD/Data_raw/ml-1m/users.dat -------------------------------------------------------------------------------- /Dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrzaurin/neural_cf/HEAD/Dataset.py -------------------------------------------------------------------------------- /GMF_gluon.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrzaurin/neural_cf/HEAD/GMF_gluon.py -------------------------------------------------------------------------------- /GMF_keras.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrzaurin/neural_cf/HEAD/GMF_keras.py -------------------------------------------------------------------------------- /GMF_pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrzaurin/neural_cf/HEAD/GMF_pytorch.py -------------------------------------------------------------------------------- /MLP_gluon.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrzaurin/neural_cf/HEAD/MLP_gluon.py -------------------------------------------------------------------------------- /MLP_keras.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrzaurin/neural_cf/HEAD/MLP_keras.py -------------------------------------------------------------------------------- /MLP_pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrzaurin/neural_cf/HEAD/MLP_pytorch.py -------------------------------------------------------------------------------- /NeuMF_gluon.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrzaurin/neural_cf/HEAD/NeuMF_gluon.py -------------------------------------------------------------------------------- /NeuMF_keras.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrzaurin/neural_cf/HEAD/NeuMF_keras.py -------------------------------------------------------------------------------- /NeuMF_pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrzaurin/neural_cf/HEAD/NeuMF_pytorch.py -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrzaurin/neural_cf/HEAD/README.md -------------------------------------------------------------------------------- /data_comparison.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrzaurin/neural_cf/HEAD/data_comparison.ipynb -------------------------------------------------------------------------------- /data_preparation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrzaurin/neural_cf/HEAD/data_preparation.py -------------------------------------------------------------------------------- /docs/images/GMF_model.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrzaurin/neural_cf/HEAD/docs/images/GMF_model.png -------------------------------------------------------------------------------- /docs/images/MLP_model.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrzaurin/neural_cf/HEAD/docs/images/MLP_model.png -------------------------------------------------------------------------------- /docs/images/NeuMF_model.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrzaurin/neural_cf/HEAD/docs/images/NeuMF_model.png -------------------------------------------------------------------------------- /docs/images/metrics_vs_n_emb.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrzaurin/neural_cf/HEAD/docs/images/metrics_vs_n_emb.png -------------------------------------------------------------------------------- /docs/images/train_time_vs_n_emb.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrzaurin/neural_cf/HEAD/docs/images/train_time_vs_n_emb.png -------------------------------------------------------------------------------- /evaluate_keras.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrzaurin/neural_cf/HEAD/evaluate_keras.py -------------------------------------------------------------------------------- /models/gluon_GMF_bs_1024_lr_0001_n_emb_32.params: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrzaurin/neural_cf/HEAD/models/gluon_GMF_bs_1024_lr_0001_n_emb_32.params -------------------------------------------------------------------------------- /models/gluon_MLP_bs_512_reg_00_lr_001_n_emb_128_ll_64_dp_wodp.params: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrzaurin/neural_cf/HEAD/models/gluon_MLP_bs_512_reg_00_lr_001_n_emb_128_ll_64_dp_wodp.params -------------------------------------------------------------------------------- /models/keras_GMF_bs_1024_reg_00_lr_0001_n_emb_16.h5: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrzaurin/neural_cf/HEAD/models/keras_GMF_bs_1024_reg_00_lr_0001_n_emb_16.h5 -------------------------------------------------------------------------------- /models/keras_MLP_bs_256_reg_00_lr_0001_n_emb_128_ll_64_dp_wodp.h5: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrzaurin/neural_cf/HEAD/models/keras_MLP_bs_256_reg_00_lr_0001_n_emb_128_ll_64_dp_wodp.h5 -------------------------------------------------------------------------------- /models/pytorch_GMF_bs_256_lr_0001_n_emb_32.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrzaurin/neural_cf/HEAD/models/pytorch_GMF_bs_256_lr_0001_n_emb_32.pt -------------------------------------------------------------------------------- /models/pytorch_MLP_bs_256_reg_00_lr_001_n_emb_64_ll_32_dp_wodp.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrzaurin/neural_cf/HEAD/models/pytorch_MLP_bs_256_reg_00_lr_001_n_emb_64_ll_32_dp_wodp.pt -------------------------------------------------------------------------------- /models/results_df.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrzaurin/neural_cf/HEAD/models/results_df.csv -------------------------------------------------------------------------------- /plot_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrzaurin/neural_cf/HEAD/plot_utils.py -------------------------------------------------------------------------------- /results_summary.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrzaurin/neural_cf/HEAD/results_summary.ipynb -------------------------------------------------------------------------------- /run_net.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrzaurin/neural_cf/HEAD/run_net.sh -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jrzaurin/neural_cf/HEAD/utils.py --------------------------------------------------------------------------------