Прогнозирование машинного обучения с отображением вероятности и целевого значения

Question

Я создал мульти-классифицирующую модель SV C, которая на основе текста классифицирует входной текст по одному из ~ 100 целевых значений.

Мне удалось отобразить Predict_proba:

Вывод:

array([[0.050, 0.001, 0.002, 0.701, 0.032, 0.008, 0.002, 0.001, 0.002,
        0.003, 0.005, 0.001 .... ]])

Однако я также хочу, чтобы вывод отображал только то, к какой целевой переменной он принадлежит. Я также хочу, чтобы он сначала выводил целевую переменную с наибольшей вероятностью. Это может выглядеть примерно так:

Вывод: Target4: 0.701, Target1: 0.050, Target5: 0.032 et c .....

Как я могу это сделать?

Также было бы замечательно отображать только 10 лучших прогнозируемых целевых значений (поэтому я не получаю выходных данных, состоящих из вероятности ~ 100 целевых значений).

Answer 1

Если вы используете sklearn, вы можете попробовать этот фрагмент, чтобы распечатать списки верхних N классов и их соответствующих вероятностей:

print(sorted(zip(clf.classes_, probs[0]), key=lambda x:x[1])[-N:])