Shortcuts

unike.module.loss

损失函数部分。

损失函数基类

Loss

继承自 unike.module.BaseModule,什么额外的属性都没有增加。

损失函数子类

MarginLoss

TransE [BUGD+13] 原论文中应用这种损失函数完成模型训练。

SigmoidLoss

RotatE [SDNT19] 原论文中应用这种损失函数完成模型训练。

SoftplusLoss

ComplEx [TWR+16] 原论文中应用这种损失函数完成模型训练。

RGCNLoss

R-GCN [SKB+18] 原论文中应用这种损失函数完成模型训练。

CompGCNLoss

CompGCN [VSNT20] 原论文中应用这种损失函数完成模型训练。

超参数优化默认搜索范围

get_margin_loss_hpo_config

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

get_sigmoid_loss_hpo_config

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

get_softplus_loss_hpo_config

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

get_rgcn_loss_hpo_config

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

get_compgcn_loss_hpo_config

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

Docs

Access comprehensive developer documentation for UniKE

View Docs