clf_gbdt = GradientBoostingClassifier()
clf_gbdt.fit(x_train, y_train)
y_preds_gbdt = clf_gbdt.predict(x_test)
y_preds_proba = clf_gbdt.predict_proba(x_test)
print('fi_score:',f1_score(y_test,y_preds_gbdt))
print(y_preds_proba[:,1])
print(y_preds_proba[:,1].shape,y_test.shape)
print('roc_auc_score:',roc_auc_score(y_test,y_preds_proba[:,1]))
y_preds_proba[:,1]
最新推荐文章于 2024-03-06 16:35:06 发布