前置代码:(分数据就不多赘述了)
# 定义RF
rf = RandomForestClassifier(n_estimators=40, max_depth=5, random_state=0)
# 训练模型
rf_SDVData2.fit(X_train, Y_train)
# predict_proba()返回预测属于某标签的概率
pred = rf.predict_proba(X_test)[:, 1]
方法一
# roc_auc_score函数返回曲线下面积
roc_auc = roc_auc_score(Y_test, pred)
方法二
fpr_Nb, tpr_Nb, _ = roc_curve(df_data_sdv_Y_test, pred)
roc_auc = auc(fpr_Nb, tpr_Nb)