注意-GRU-3M
论文“具有自适应Attention-GRU模型的品牌级排名系统[C]”的代码。 (接受IJCAI 2018)
运行命令:python train.py --buckets“ ./data/” --checkpointDir ./log/ --exp debug --m1 1 --m2 0 --m3 1 参数:“ buckets”是输入数据的文件夹,“ checkpointDir”是输出数据的折叠,“ debug”仅是文件夹名称(不重要),“ m1 m2 m3”表示激活了哪个修改。
运行上述命令一段时间后,在目录根目录下运行:tensorboard --logdir =“ ./ log /”,然后可以通过网站链接查看性能。
虚拟环境:virtualenv -p python2 env 源环境/ bin /激活
在Linux上安装TensorFlow: pip安装-
2021-10-18 21:46:52
10KB
Python
1