Getting Started
Bring Your Own
Extending PyKEEN
Reference
pykeen.nn
Appendix
Create a dataframe with relation counts.
dataset (Dataset) – The dataset.
add_labels (bool) – Whether to add relation labels to the dataframe.
merge_subsets (bool) – Whether to merge subsets, i.e., train/validation/test.
add_labels – Whether to add entity / relation labels.
A dataframe with columns (relation_id, count, relation_label?, subset?)
DataFrame