from sklearn.tree import DecisionTreeClassifier from sklearn.metrics import roc_auc_score model = DecisionTreeClassifier() model.fit(x_train, y_train) y_pred = model.predict_proba(x_train)[:, 1] train_auc = roc_auc_score(y_train, y_pred) print('train_auc:', train_auc) # 1.0 y_pred = model.predict_proba(x_val)[:, 1] val_auc = roc_auc_score(y_val, y_pred) print('val_auc:', val_auc) # 0.61 __ __