Shortcuts

pybind11_ke.data

数据采样部分,包含为训练和验证模型定义的数据采样器。

数据预加载器

KGReader

从文件中读取知识图谱。

平移模型和语义匹配模型训练集数据采样器

TradSampler

平移模型和语义匹配模型的采样器的基类。

UniSampler

平移模型和语义匹配模型的训练集普通的数据采样器(均值分布)。

BernSampler

平移模型和语义匹配模型的训练集 Bern 数据采样器(伯努利分布),如果想获得更详细的信息请访问 TransH

图神经网络模型训练集数据采样器

RevSampler

在训练集中增加相反关系.

RGCNSampler

R-GCN [SKB+18] 的训练数据采样器。

CompGCNSampler

CompGCN [VSNT20] 的训练数据采样器。

测试集数据采样器

TestSampler

测试数据采样器基类。

TradTestSampler

平移模型和语义匹配模型的测试数据采样器。

RGCNTestSampler

R-GCN [SKB+18] 的测试数据采样器。

CompGCNTestSampler

CompGCN [VSNT20] 的测试数据采样器。

数据加载器

KGEDataLoader

KGE 模型数据加载器。

超参数优化默认搜索范围

get_kge_data_loader_hpo_config

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

Docs

Access comprehensive developer documentation for Pybind11-OpenKE

View Docs