X_train,X_test,y_train,y_test=train_test_split(X ,y,test_size=0.2,random_state=1) from sklearn import tree DecTreeModel = tree.DecisionTreeClassifier().fit(X_train, y_train) preds = DecTreeModel.predict(X_test) print('accuracy', metrics.accuracy_score(y_test, preds)) print('recall', metrics.recall_score(y_test, preds)) print('precision', metrics.precision_score(y_test, preds)) print('f1-score', metrics.f1_score(y_test, preds)) print(classification_report(y_test, preds)) y_scores = DecTreeModel.predict_proba(X_test) from sklearn.metrics import roc_curve from sklearn.metrics import confusion_matrix import matplotlib import matplotlib.pyplot as plt %matplotlib inline # calculate ROC curve fpr, tpr, thresholds = roc_curve(y_test, y_scores[:,1]) # plot ROC curve fig = plt.figure(figsize=(16, 8)) # Plot the diagonal 50% line plt.plot([0, 1], [0, 1], 'k--') # Plot the FPR and TPR achieved by our model plt.plot(fpr, tpr) plt.xlabel('False Positive Rate') plt.ylabel('True Positive Rate') plt.title('ROC Curve') plt.show()

这段代码实现了一个基于决策树的分类器，并对其性能进行了评估和可视化分析。首先，使用 train_test_split 函数将数据集分为训练集和测试集，然后使用决策树分类器进行训练，预测测试集上的结果并计算准确率、召回率、精确度和 F1 得分等指标。接着，使用 predict_proba 函数计算测试集上的预测得分，然后使用 roc_curve 函数计算接收器操作特征曲线（ROC 曲线）的 FPR 和 TPR，最后使用 Matplotlib 库绘制 ROC 曲线，并且将其展示出来。