Shortcuts

unike.module.strategy

训练策略部分。

策略基类

Strategy

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

策略子类

NegativeSampling

将模型和损失函数封装到一起,方便模型训练。

RGCNSampling

将模型和损失函数封装到一起,方便模型训练,用于 R-GCN [SKB+18]

CompGCNSampling

将模型和损失函数封装到一起,方便模型训练,用于 CompGCN [VSNT20]

超参数优化默认搜索范围

get_negative_sampling_hpo_config

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

get_rgcn_sampling_hpo_config

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

get_compgcn_sampling_hpo_config

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

Docs

Access comprehensive developer documentation for UniKE

View Docs