TDDPG-Rec
该代码可重现 (在第24届欧洲人工智能大会上,ECAI 2020)的实验结果。
数据集
数据预处理代码也包括在内。您可以从下载Amazon数据,然后运行amazon.py。
运行环境
该代码已在Windows 10和Ubuntu 16.04以及TensorFlow 1.14.0和Python 3.6.8下进行了测试。
当CUDA可用时,支持使用CPU进行独立培训以及使用CPU和GPU进行联合培训。
资源资源
您可以在./resource文件夹下下载这些资源并将其添加到该项目中。
预先训练的词向量可在上,该工具已在Wikipedia2014和Gigaword 5上进行了训练。
长停用词列表可以在获得。
模型训练
以Digital_Music为例。获取源数据后,应首先运行数据处理:
python amazon.py
要培养我们的DDPG模型上Digital_Music :
评论0
最新资源