search

ShulinCao / OpenNRE-PyTorch

Neural Relation Extraction implemented in PyTorch
MIT License
219 stars 45 forks source link

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

Open rivercold opened 4 years ago

rivercold commented 4 years ago

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

谢谢！

ErxinYu commented 4 years ago

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

pr_curve的副本

rivercold commented 4 years ago

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

ErxinYu commented 4 years ago

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

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

rivercold commented 4 years ago

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

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

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