pybind11_ke.config¶
循环部分,包含训练循环和验证循环。
训练循环¶
主要用于 KGE 模型的训练。 |
评估循环¶
主要用于 KGE 模型的评估。 |
链接预测函数¶
进行链接预测。 |
|
进行头实体的链接预测。 |
|
进行尾实体的链接预测。 |
|
计算三元组的排名。 |
并行训练函数¶
由于分布式并行依赖于 accelerate ,因此,需要利用 Accelerator 为分布式训练准备对象。 |
超参数优化默认搜索范围¶
返回 |
|
返回 |
超参数优化训练循环函数¶
设置超参数优化范围。 |
|
设置 Hits 指标。 |
|
开启超参数优化。 |
|
超参数优化训练循环函数。 |