数据挖掘项目（四）

from sklearn.linear_model import LogisticRegression
from sklearn.svm import SVC, LinearSVC
from sklearn.ensemble import RandomForestClassifier
from sklearn.tree import DecisionTreeClassifier
from xgboost import XGBClassifier

models = [LogisticRegression(), LinearSVC(), DecisionTreeClassifier(), RandomForestClassifier(n_estimators=25), XGBClassifier()]
names = ['LogisticRegression','LinearSVC','DecisionTree','RandomForest','Xgboost']
for i, model in enumerate(models):
    model.fit(X_selected, y_train)
    accu = model.score(X_test_selected, y_test)
    print(names[i], accu)

from sklearn.metrics import roc_curve, auc

def plot_auc(model_name, y_pred, y_test):
    fpr, tpr, threshold = roc_curve(y_pred, y_test)
    roc_auc = auc(fpr, tpr)
    plt.plot(fpr, tpr, label=f'{names[i]}（area={roc_auc:.2f}）', )
    plt.pl