欢迎来到 UniKE 文档！¶ UniKE 是一个知识图谱嵌入工具包，能够运行在 Windows 和 Linux 操作系统上。 为了使用 UniKE，请先 安装 这个项目。 目录¶ 概述 主页 安装 数据集 评估指标 多 GPU 配置 实现细节 教程 知识图谱嵌入模型 API API unike.utils 时间类 Timer 日志类 WandbLogger 早停止类 EarlyStopping 链接预测类 Link 导入模块类函数 import_class construct_type_constrain unike.data 数据预加载器 KGReader 平移模型和语义匹配模型训练集数据采样器 TradSampler UniSampler BernSampler 图神经网络模型训练集数据采样器 RevSampler RGCNSampler CompGCNSampler 测试集数据采样器 TestSampler TradTestSampler RGCNTestSampler CompGCNTestSampler 数据加载器 KGEDataLoader 超参数优化默认搜索范围 get_kge_data_loader_hpo_config unike.module 基础模块 BaseModule unike.module.model 基础模块 CompGCNCov 模型基类 Model 平移模型 TransE TransH TransR TransD RotatE 语义匹配模型 RESCAL DistMult HolE ComplEx Analogy SimplE 图神经网络模型 RGCN CompGCN 平移模型超参数优化默认搜索范围 get_transe_hpo_config get_transh_hpo_config get_transr_hpo_config get_transd_hpo_config get_rotate_hpo_config 语义匹配模型超参数优化默认搜索范围 get_rescal_hpo_config get_distmult_hpo_config get_hole_hpo_config get_complex_hpo_config get_analogy_hpo_config get_simple_hpo_config 图神经网络模型超参数优化默认搜索范围 get_rgcn_hpo_config get_compgcn_hpo_config unike.module.loss 损失函数基类 Loss 损失函数子类 MarginLoss SigmoidLoss SoftplusLoss RGCNLoss CompGCNLoss 超参数优化默认搜索范围 get_margin_loss_hpo_config get_sigmoid_loss_hpo_config get_softplus_loss_hpo_config get_rgcn_loss_hpo_config get_compgcn_loss_hpo_config unike.module.strategy 策略基类 Strategy 策略子类 NegativeSampling RGCNSampling CompGCNSampling 超参数优化默认搜索范围 get_negative_sampling_hpo_config get_rgcn_sampling_hpo_config get_compgcn_sampling_hpo_config unike.config 训练循环 Trainer 评估循环 Tester 链接预测函数 link_predict head_predict tail_predict calc_ranks 超参数优化默认搜索范围 get_trainer_hpo_config get_tester_hpo_config 超参数优化训练循环函数 set_hpo_config set_hpo_hits start_hpo_train hpo_train Indices and tables 参考 参考文献