get_entity_count_df

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

创建一个包含实体计数的数据框。

Parameters:

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

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

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

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

Returns:

每个实体一行数据的数据框。

Return type:

数据框