From f53e3f2dba140ef119f22540bb7fdd16494b37b9 Mon Sep 17 00:00:00 2001 From: pseuc2htk <1085670763@qq.com> Date: Sat, 26 Jun 2021 10:41:16 +0800 Subject: [PATCH] Update README.md --- README.md | 18 +++++++++++++++++- 1 file changed, 17 insertions(+), 1 deletion(-) diff --git a/README.md b/README.md index 56bc505..65fe170 100644 --- a/README.md +++ b/README.md @@ -1 +1,17 @@ -基于pytorch的NeuralCF实现电影评分预测及推荐 \ No newline at end of file +### 功能说明 +基于pytorch的NeuralCF实现电影评分预测及推荐 +### 使用说明 +1. 环境配置: +> python +pytorch +sklearn +numpy +pandas + +2. 路径设置 +设置`./tool/train_test.py`文件下路径,保证能正常访问数据。 + +3. 模型训练和推理 +在`./tool`文件夹下,配置`./tool/train_test.py`下训练参数后,运行: +`python train_test.py` +进行训练和推理。