获取实体关系共现数据框

get_entity_relation_co_occurrence_df(dataset: Dataset, merge_sides: bool = True, merge_subsets: bool = True, add_labels: bool = True) → DataFrame[source]

创建一个实体/关系共现的数据框。

这些信息可以被视为一种伪类型的形式，例如实体A可以是born_in的头部。

Parameters:

  • 数据集 (Dataset) – 数据集。

  • merge_sides (bool) – 是否合并两侧，即实体位置：头 vs. 尾。

  • merge_subsets (bool) – 是否合并子集，即训练/验证/测试。

  • add_labels (bool) – 是否添加实体/关系标签。

Returns:

一个包含实体-关系对及其出现次数的数据框。

Return type:

数据框