PyTorch实现似乎比原paper的结果好？

[rivercold]

我发现这个PyTorch的实现能取得比PCNN-ATT Paper里显著更好的 P@k 和 PR-Curves。不知道你跑的时候是不是也是这样？

谢谢！

[ErxinYu]

你是用的json格式的那个数据集么(train_instance:522611) 为什么我跑出来的pcnn+ATT效果还不如paper中报告的。

[rivercold]

是的，520k 的 training json，先跑了gen_data.py。
我主要是P@K比较高。
你optimization的参数是什么？

[ErxinYu]

是的，520k 的 training json，先跑了gen_data.py。
我主要是P@K比较高。
你optimization的参数是什么？

你好，我是按照代码中的参数进行优化，但是和论文中不同的是代码中的学习率是0.5，而论文中标的是0.01. 请问你是怎么处理的呢

[rivercold]

是的，520k 的 training json，先跑了gen_data.py。
我主要是P@K比较高。
你optimization的参数是什么？

你好，我是按照代码中的参数进行优化，但是和论文中不同的是代码中的学习率是0.5，而论文中标的是0.01. 请问你是怎么处理的呢

试试 SGD 用 lr =0. 1 然后可以把weight_decay设为0看看。