Блог

Главная - Вопросы по программированию - Как изменить легенды сюжета с помощью roc_auc_score?

Как изменить легенды сюжета с помощью roc_auc_score?

#python #plot #scikit-learn #roc #auc

#python #сюжет #scikit-learn #ОКР #auc

Вопрос:

Я строю кривую ROC с plot_roc_curve помощью of scikit-learn , чтобы легенды сюжета печатались автоматически. Есть ли способ их изменить?

 metrics.plot_roc_curve(classifier, X_test, y_test, ax=plt.gca())

Ответ №1:

Вы можете в значительной степени добавить все, что вам нравится, к объекту сюжета, который создается с помощью plot_roc_curve . Например, вы можете сделать что-то вроде этого:

 plot_roc_curve(classifier, X_test, y_test, ax=plt.gca(), label="Legend")
plt.xlabel("Your x label text")
plt.ylabel("Your y label text")
plt.title("Some title")

Это вернет:

Кривая ROC

Комментарии:

1. Спасибо. Могу ли я отредактировать часть автоматически сгенерированных легенд и сохранить счет? Например, преобразовать LogisticRegression (AUC = 0.82) в LR (AUC = 0.82) .

2. @Flodude попробуйте name='LR' .