cksruf91/recomm-deepfm-pytorch

recomm-deepfm-pytorch

Dataset

• MovieLens 1M Dataset

cd {project Dir}/datasets/movielens
wget https://files.grouplens.org/datasets/movielens/ml-1m.zip
unzip ml-1m.zip

• MovieLens 10M Dataset

cd {project Dir}/datasets/movielens
wget https://files.grouplens.org/datasets/movielens/ml-10m.zip
unzip ml-10m.zip

data preprocess

python preprocess.py -d 1M

train model

python deepfm_train.py -d 1M -v 0.1.0 -k 10 -ed 16 -lr 0.001 -bs 256 -ly 64 32 -o 0.2