from sklearn.linear_model import LogisticRegression from sklearn.metrics import roc_auc_score model = LogisticRegression() 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) # 0.61 __ __