predict

predict(model, *, k=None, batch_size=1)[source]

Calculate and store scores for either all triples, or the top k triples.

Parameters

  • model (Model) – A PyKEEN model

  • k (Optional[int]) – The number of triples to return. Set to None to keep all.

  • batch_size (int) – The batch size to use for calculating scores

Return type

ScorePack

Returns

A score pack of parallel triples and scores