- predict_all(model, *, k=None, batch_size=1, mode=None, target='tail')
Calculate scores for all triples, and either keep all of them or only the top k triples.
Model) – A PyKEEN model
Literal[‘head’, ‘relation’, ‘tail’]) – the prediction target to use. Prefer targets which are efficient to predict with the given model, e.g., tails for ConvE.
- Return type:
A score pack of parallel triples and scores
ValueError – if an inductive inference mode is selected, but the model does not support it