OpenKE中NELL-995数据集的neg_ent设置指南

在使用OpenKE框架进行知识表示学习时,正确设置负样本数量至关重要。对于NELL-995数据集,官方文档建议负样本比例为0.5,这意味着负样本数量应该等于正样本数量。

由于NELL-995数据集中每个实体有两个关系,因此每个实体拥有两个正样本。所以,在OpenKE中配置NELL-995数据集时,neg_ent参数应该设置为1。

总结:

  • NELL-995数据集的推荐负样本比例:0.5
  • 每个实体的正样本数量:2
  • neg_ent设置:1

通过正确设置neg_ent,可以确保模型在训练过程中获得均衡的正负样本,从而提高知识表示学习的效果。

OpenKE中NELL-995数据集的neg_ent设置指南

原文地址: https://www.cveoy.top/t/topic/jnUC 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录