rank_score¶
-
ampligraph.evaluation.
rank_score
(y_true, y_pred, pos_lab=1)¶ Rank of a triple
The rank of a positive element against a list of negatives.\[rank_{(s, p, o)_i}\]Parameters: - y_true (ndarray, shape [n]) – An array of binary labels. The array only contains one positive.
- y_pred (ndarray, shape [n]) – An array of scores, for the positive element and the n-1 negatives.
- pos_lab (int) – The value of the positive label (default = 1).
Returns: rank – The rank of the positive element against the negatives.
Return type: int
Examples
>>> import numpy as np >>> from ampligraph.evaluation.metrics import rank_score >>> y_pred = np.array([.434, .65, .21, .84]) >>> y_true = np.array([0, 0, 1, 0]) >>> rank_score(y_true, y_pred) 4