Shortcuts

unike.config

循环部分,包含训练循环和验证循环。

训练循环

Trainer

主要用于 KGE 模型的训练。

评估循环

Tester

主要用于 KGE 模型的评估。

链接预测函数

link_predict

进行链接预测。

head_predict

进行头实体的链接预测。

tail_predict

进行尾实体的链接预测。

calc_ranks

计算三元组的排名。

超参数优化默认搜索范围

get_trainer_hpo_config

返回 Trainer 的默认超参数优化配置。

get_tester_hpo_config

返回 Tester 的默认超参数优化配置。

超参数优化训练循环函数

set_hpo_config

设置超参数优化范围。

set_hpo_hits

设置 Hits 指标。

start_hpo_train

开启超参数优化。

hpo_train

超参数优化训练循环函数。

Docs

Access comprehensive developer documentation for UniKE

View Docs