sklearn中predict和predict_proba的区别和使用
1.predict与predict_proba区别
- 都用于模型的预测
- predict返回的是预测的值(二分类则是0,1),predict——proba返回的是预测各个类别的概率。
- predict_proba返回的是一个n行k列的数组,n表示测试集中样本的个数,地i行j列数值是模型预测第i个预测样本某个标签的概率,每行之和为1.相对更精准。比如在画ROC图使用。
2.举例说明
xgb.XGBClassifier(max_depth=6,n_estimators=100).fit(X_train,y_train).predict(X_test)
xgb.XGBClassifier(max_depth=6,n_estimators=100).fit(X_train,y_train).predict_proba(X_test)
No pains, No gains