Skip to content

Commit 7eda20a

Browse files
committedAug 16, 2024
deprecated get_tail_prediction_df still in docs
1 parent 0f26891 commit 7eda20a

File tree

1 file changed

+5
-6
lines changed

1 file changed

+5
-6
lines changed
 

‎docs/source/tutorial/understanding_evaluation.rst

+5-6
Original file line numberDiff line numberDiff line change
@@ -53,15 +53,14 @@ We can see the results using the prediction workflow:
5353

5454
.. code-block:: python
5555
56-
from pykeen.models.predict import get_tail_prediction_df
56+
from pykeen.predict import predict_target
5757
58-
df = get_tail_prediction_df(
58+
df = predict_target(
5959
model=result.model,
60-
head_label="belgium",
61-
relation_label="locatedin",
60+
tail="belgium",
61+
relation="locatedin",
6262
triples_factory=result.training,
63-
add_novelties=False,
64-
)
63+
).df
6564
6665
which returns a dataframe of all tail candidate entities sorted according to the predicted score.
6766
The index in this sorted list is essentially the *rank* of the correct answer.

0 commit comments

Comments
 (0)
Please sign in to comment.