Shortcuts

pybind11_ke.data

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

平移模型和语义匹配模型数据加载器

TrainDataSampler

pybind11_ke.data.TrainDataLoader.sampling()pybind11_ke.data.TrainDataLoader.cross_sampling() 进行封装。

TrainDataLoader

主要从底层 C++ 模块获得数据用于 KGE 模型的训练。

TestDataSampler

pybind11_ke.data.TestDataLoader.sampling() 进行封装。

TestDataLoader

主要从底层 C++ 模块获得数据用于 KGE 模型的评估。

图神经网络模型数据加载器

KGReader

从文件中读取知识图谱。

RevSampler

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

GraphSampler

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

CompGCNSampler

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

GraphTestSampler

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

CompGCNTestSampler

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

GraphDataLoader

基本图神经网络采样器。

超参数优化默认搜索范围

get_train_data_loader_hpo_config

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

get_test_data_loader_hpo_config

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

get_graph_data_loader_hpo_config

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

Docs

Access comprehensive developer documentation for Pybind11-OpenKE

View Docs