根据 OpenKE 官方文档,NELL-995 数据集的 neg_ent 参数默认设置为 25。但是可以根据实际需要进行调整。

neg_ent 参数表示在训练过程中每个正样本需要采样的负样本数量。默认值为 25 表示每个正样本会随机选择 25 个负样本进行训练。

调整 neg_ent 参数需要根据具体情况进行考虑。

  • 如果负样本数量过多,会导致训练时间过长,但可能会提高模型的性能。
  • 如果负样本数量过少,会导致模型训练不足,但可能会缩短训练时间。

建议根据数据量、模型复杂度以及训练资源等因素综合考虑,并进行实验验证找到最佳的 neg_ent 参数设置。

OpenKE NELL-995 数据集 neg_ent 参数设置

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

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