Как вычислить среднюю точность из ответа ранжированного списка на запрос, используя sci-kit learn average_precision_score?

Я хочу вычислить показатель средней средней точности моей поисковой системы. Я написал скрипт для вычисления средней точности по формуле, упомянутой в https://en.wikipedia.org/wiki/Information_retrieval.

Но я хочу вычислить это с помощью функции sci-kit learn average_precision_score, чтобы сравнить и сравнить мою реализацию.

from sklearn.metrics import average_precision_score

#suppose i have a response of ranked list of docs,let's say [doc1, doc2, doc3, doc4]

#I compute the ground truth from this ranked-list. (where the last two docs are wrongly retrieved items)
ground_truth = [1, 1, 0, 0]

target_scores = ?

#so the average precision should be able to computed using sklearn function like below
average_precision_score(y_true = ground_truth, y_score= target_scores , average="samples")

До сих пор у меня нет никакого представления о том, как вычислить этот параметр 'y_score', необходимый для функции 'average_precision_score', из имеющегося у меня списка ground_truth.

0 ответов

Другие вопросы по тегам