1

我正在尝试scikitplot.metrics.plot_calibration_curve为我的模型绘制校准曲线,并希望更改结果图表中的线型(例如虚线、实线、点线)。

我可以制作的最简单的可重现示例如下。

import scikitplot as skplt
from sklearn.ensemble import RandomForestClassifier
from sklearn.linear_model import LogisticRegression
from sklearn.datasets import load_breast_cancer
from sklearn.model_selection import train_test_split

# load the breast_cancer dataset and split it into train and test sets
X, y = load_breast_cancer(return_X_y=True)
X_train, X_test, y_train, y_test = train_test_split(X, y, random_state=0)

rf = RandomForestClassifier()
lr = LogisticRegression()
rf_probas = rf.fit(X_train, y_train).predict_proba(X_test)
lr_probas = lr.fit(X_train, y_train).predict_proba(X_test)
clf_names = ['Random Forest', 'Logistic Regression']
probas_list = [rf_probas, lr_probas]
skplt.metrics.plot_calibration_curve(y_test,
                                      probas_list,
                                      clf_names)

这正是我想要的:

在此处输入图像描述

但我只想能够更改线型,以便图表可以黑白打印。

4

0 回答 0