We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 0f26891 commit 7eda20aCopy full SHA for 7eda20a
docs/source/tutorial/understanding_evaluation.rst
@@ -53,15 +53,14 @@ We can see the results using the prediction workflow:
53
54
.. code-block:: python
55
56
- from pykeen.models.predict import get_tail_prediction_df
+ from pykeen.predict import predict_target
57
58
- df = get_tail_prediction_df(
+ df = predict_target(
59
model=result.model,
60
- head_label="belgium",
61
- relation_label="locatedin",
+ tail="belgium",
+ relation="locatedin",
62
triples_factory=result.training,
63
- add_novelties=False,
64
- )
+ ).df
65
66
which returns a dataframe of all tail candidate entities sorted according to the predicted score.
67
The index in this sorted list is essentially the *rank* of the correct answer.
0 commit comments