Prediction workflows.


predict(model, *[, k, batch_size])

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

get_all_prediction_df(model, *[, k, …])

Compute scores for all triples, optionally returning only the k highest scoring.

get_head_prediction_df(model, …[, …])

Predict heads for the given relation and tail (given by label).

get_relation_prediction_df(model, …[, …])

Predict relations for the given head and tail (given by label).

get_tail_prediction_df(model, head_label, …)

Predict tails for the given head and relation (given by label).