pseuc2htk
/
recommendation_algorithm_using_deeplearning
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
11 Commits
1 Branch
0 Tags
21 MiB
 
Branches Tags
${ item.name }
Create tag ${ searchTerm }
Create branch ${ searchTerm }
from '6040629bee'
${ noResults }
Go to file
pseuc2htk 6040629bee
Update README.md 		3 years ago
AutoRec first commit 3 years ago
Data/ml-1m first commit 3 years ago
TrainedModels first commit 3 years ago
tool first commit 3 years ago
.gitignore Initial commit 3 years ago
LICENSE Initial commit 3 years ago
README.md Update README.md 3 years ago
fea_col.npy 数据 3 years ago

README.md

功能说明

基于pytorch的AutoRec实现电影评分预测及推荐

使用说明

  1. 下载: 使用git clone https://bdgit.educoder.net/pseuc2htk/recommendation_algorithm_using_deeplearning.git命令下载,或者直接下载zip包解压使用。
  2. 环境配置： 要求python>=3.6,并配置好GPU环境,其他环境配置如下：

pytorch>=1.4 cuda=10.0 cuDNN =7.6 sklearn numpy pandas

  1. 路径设置 设置./tool/train_test.py文件下路径，保证能正常访问数据。

  2. 模型训练和推理 在./tool文件夹下，配置./tool/train_test.py下训练参数后，运行：
    python train_test.py
    进行训练和推理。

  3. 运行train_test.py中的推理部分，对给出的测试数据进行预测并打印出预测值。

  4. 将打印出的预测值粘贴到测试平台的result/result.txt中进行评测；