Prediction

Prediction workflows.

Functions

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

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

predict_triples_df(model, *, triples[, ...])

Predict on labeled or mapped triples.

get_all_prediction_df(model, *, triples_factory)

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).