get_relation_count_df

get_relation_count_df(dataset: Dataset, merge_subsets: bool = True, add_labels: bool = True) → DataFrame

创建一个包含关系计数的数据框。

Parameters:

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

• add_labels (bool) – 是否向数据框添加关系标签。

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

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

Returns:

一个包含列（relation_id, count, relation_label?, subset?）的数据框

Return type:

数据框