parent
dc709da727
commit
f53e3f2dba
@ -1 +1,17 @@
|
||||
基于pytorch的NeuralCF实现电影评分预测及推荐
|
||||
### 功能说明
|
||||
基于pytorch的NeuralCF实现电影评分预测及推荐
|
||||
### 使用说明
|
||||
1. 环境配置:
|
||||
> python
|
||||
pytorch
|
||||
sklearn
|
||||
numpy
|
||||
pandas
|
||||
|
||||
2. 路径设置
|
||||
设置`./tool/train_test.py`文件下路径,保证能正常访问数据。
|
||||
|
||||
3. 模型训练和推理
|
||||
在`./tool`文件夹下,配置`./tool/train_test.py`下训练参数后,运行:
|
||||
`python train_test.py`
|
||||
进行训练和推理。
|
||||
|
Loading…
Reference in new issue