You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
pseuc2htk f53e3f2dba
Update README.md
3 years ago
Data/ml-1m first commit 3 years ago
NCF first commit 3 years ago
TrainedModels first commit 3 years ago
tool first commit 3 years ago
.gitignore Initial commit 3 years ago
LICENSE Initial commit 3 years ago
README.md Update README.md 3 years ago

README.md

功能说明

基于pytorch的NeuralCF实现电影评分预测及推荐

使用说明

  1. 环境配置:

python pytorch sklearn numpy pandas

  1. 路径设置 设置./tool/train_test.py文件下路径,保证能正常访问数据。

  2. 模型训练和推理 在./tool文件夹下,配置./tool/train_test.py下训练参数后,运行:
    python train_test.py
    进行训练和推理。