После многочисленных попыток я не могу понять, как можно восстановить метрики классов от precision_recall_fscore_support
возвращаемых значений.Не удается определить упорядочение классов в значениях возвращаемых значений precision_recall_fscore_support в scikit-learn
Например, учитывая этот классический контекст обучения:
target_names = set(y)
y = [target_names.index(x) for x in y]
X_train, X_test, y_train, y_test = train_test_split(X, y)
# Some classification ...
y_pred = clf.predict(X_test)
precision, recall, f1, support = precision_recall_fscore_support(y_test, y_pred)
Здесь len(set(y_test)) == len(support)
так я представляю себе, что все классы, присутствующие в y_test
присутствуют в возвращаемых значений. Но я не могу найти способ, которым они упорядочены, поэтому я могу восстановить, какие показатели соответствуют классу.
Благодарим за помощь!