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` +进行训练和推理。