- predict_target(model, *, head=None, relation=None, tail=None, triples_factory=None, targets=None, mode=None)
Get predictions for the head, relation, and/or tail combination.
Exactly one of head, relation and tail should be None. This is the position which will be predicted.
Model) – A PyKEEN model
- Return type:
The predictions, containing either the \(k\) highest scoring targets, or all targets if \(k\) is None.