OpenKE: NELL-995数据集参数设置指南 (neg_ent, neg_rel, dim)

OpenKE: NELL-995数据集参数设置指南 (neg_ent, neg_rel, dim)

在使用OpenKE框架训练NELL-995数据集时，neg_ent、neg_rel和dim是三个重要的参数，会影响模型性能和训练效率。

参数解释:

• neg_ent: 每个正例对应的负样本实体数量。
• neg_rel: 每个正例对应的负样本关系数量。
• `dim**: 嵌入向量的维度。

推荐设置:

• neg_ent和neg_rel: 建议设置为10-100之间。更大的值可以提供更强的负样本对比，但也会增加训练时间。
• dim: 建议设置为50-500之间。更大的值可以增强模型表达能力，但需要更多的计算资源和训练时间。

注意事项:

• 最佳参数设置需要根据具体的实验目标、数据集大小和计算资源进行调整。
• 可以通过网格搜索或其他超参数优化方法找到最佳参数组合。

希望这份指南可以帮助您在OpenKE中更好地使用NELL-995数据集。