Link

class unike.utils.Link(in_path: str = './', ent_file: str = 'entity2id.txt', rel_file: str = 'relation2id.txt', all_file: str = 'all2id.txt')

使用 KGE 模型对感兴趣的三元组计算链接分数。

__init__(in_path: str = './', ent_file: str = 'entity2id.txt', rel_file: str = 'relation2id.txt', all_file: str = 'all2id.txt')

创建 Link 对象。

参数:

• model (Model) – 模型

• in_path (str) – 数据集目录

• ent_file (str) – entity2id.txt

• rel_file (str) – relation2id.txt

• all_file (str) – all2id.txt

__weakref__

list of weak references to the object (if defined)

link(head_ids: list[int], rel_ids: list[int], tail_ids: list[int], model: Model, device: str = 'cpu') → pandas.DataFrame

对给定的头实体、关系和尾实体进行组合并使用模型计算链接分数。

参数:

• head_ids (list[int]) – 头实体列表

• rel_ids (list[int]) – 关系列表

• tail_ids (list[int]) – 尾实体列表

• device (str) – 模型

• device – 设备

返回:

结果数据框

返回类型:

pd.DataFrame

load_all() → None

读取 all_file 文件。

load_data() → None

读取 ent_file 文件和 rel_file 文件。