我运行与示例文件相同的代码，但绘制了另一条R O C曲线
import pandas as pd
from sklearn.model_selection import train_test_split
import yellowbrick as yb
pd.set_option('display.max_columns', None)
data_raw = pd.read_csv("data/NKI_cleaned.csv")
features_to_drop = data_raw.columns[16:]
data_subset = data_raw.drop(features_to_drop, axis=1)
X = data_subset.drop(['Patient', 'ID', 'eventdeath'], axis=1)
y = data_subset['eventdeath']
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.33, random_state=42)
from sklearn.linear_model import LogisticRegression
logreg = LogisticRegression()
logreg.fit(X_train, y_train)
logreg_pred = logreg.predict(X_test)
from sklearn.metrics import accuracy_score
logreg_acc_score = accuracy_score(y_test, logreg_pred)
print(logreg_acc_score)
from yellowbrick.classifier import ROCAUC
logreg_roc = ROCAUC(logreg)
logreg_roc.fit(X_train, y_train)
logreg_roc.score(X_test, y_test)
logreg_roc.poof()
示例文件具有一条ROC曲线。我运行的文件有四个ROC曲线。在此处输入图片描述
我用了木星笔记本。
