知识图谱补全:预测缺失的三元组链接

FreeGuideOnline 最新 2026-06-26

python from pykeen.pipeline import pipeline

定义模型与数据集,并启动训练

result = pipeline( dataset='WN18RR', model='RotatE', model_kwargs=dict(embedding_dim=512), training_kwargs=dict(num_epochs=100, batch_size=256), random_seed=42 )

查看评估指标

print(result.metric_results.to_dict())