Update README.md

master
pseuc2htk 4 years ago
parent 05d6b530bb
commit 50732bc92a

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

Loading…
Cancel
Save