Getting Started
Bring Your Own
Extending PyKEEN
Reference
pykeen.nn
Appendix
Create a dataframe with entity counts.
dataset (Dataset) – The dataset.
Dataset
merge_sides (bool) – Whether to merge sides, i.e., entity positions: head vs. tail.
bool
merge_subsets (bool) – Whether to merge subsets, i.e., train/validation/test.
add_labels (bool) – Whether to add entity / relation labels.
DataFrame
A dataframe with one row per entity.