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