get_entity_count_df

get_entity_count_df(dataset, merge_sides=True, merge_subsets=True, add_labels=True)[source]

Create a dataframe with entity counts.

Parameters
  • dataset (Dataset) – The dataset.

  • merge_sides (bool) – Whether to merge sides, i.e., entity positions: head vs. tail.

  • merge_subsets (bool) – Whether to merge subsets, i.e., train/validation/test.

  • add_labels (bool) – Whether to add entity / relation labels.

Return type

DataFrame

Returns

A dataframe with one row per entity.