Recommend Model

recommend model using pytorch

Requirements

This is my experiment eviroument

• python3.6
• pytorch1.6.0+cu101
• pandas 1.2.4

Usage

1. data

The data path in our experiment is :

• train_path: data/raw/trainingSamples.csv
• test_path: data/raw/testSamples.csv

data is from https://grouplens.org/datasets/movielens/, we sample som data from it.

2. train model

The supported model are:

1. Embedding+MLP
2. WideDeep
3. NeuralCF
4. DeepFM
5. DIN
6. DIEN

if you want to retrain model DIN, you can use command line:

python3 DIN_example.py

Implementated Network

• DeepCross(Embedding+MLP)
• WideDeep
• NeuralCF
• DeepFM
• DIN
• DIEN