parent
f53e3f2dba
commit
334daeb51f
@ -1,17 +1,21 @@
|
|||||||
### 功能说明
|
### 功能说明
|
||||||
基于pytorch的NeuralCF实现电影评分预测及推荐
|
基于pytorch的NeuralCF实现电影评分预测及推荐
|
||||||
### 使用说明
|
### 使用说明
|
||||||
1. 环境配置:
|
1. 下载:
|
||||||
> python
|
使用`git clone https://bdgit.educoder.net/pseuc2htk/NeuralCF.git`命令下载,或者直接下载zip包解压使用。
|
||||||
pytorch
|
2. 环境配置:
|
||||||
|
要求python>=3.6,并配置好GPU环境,其他环境配置如下:
|
||||||
|
> pytorch>=1.4
|
||||||
sklearn
|
sklearn
|
||||||
numpy
|
numpy
|
||||||
pandas
|
pandas
|
||||||
|
|
||||||
2. 路径设置
|
3. 路径设置
|
||||||
设置`./tool/train_test.py`文件下路径,保证能正常访问数据。
|
设置`./tool/train_test.py`文件下路径,保证能正常访问数据。
|
||||||
|
|
||||||
3. 模型训练和推理
|
4. 模型训练和推理
|
||||||
在`./tool`文件夹下,配置`./tool/train_test.py`下训练参数后,运行:
|
在`./tool`文件夹下,配置`./tool/train_test.py`下训练参数后,运行:
|
||||||
`python train_test.py`
|
`python train_test.py`
|
||||||
进行训练和推理。
|
进行训练和推理。
|
||||||
|
|
||||||
|
|
||||||
|
Loading…
Reference in new issue