sklearn中predict和predict_proba的区别和使用

1.predict与predict_proba区别

  1. 都用于模型的预测
  2. predict返回的是预测的值(二分类则是0,1),predict——proba返回的是预测各个类别的概率。
  3. 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)
posted @ 2022-05-26 16:44  Right2014  阅读(1883)  评论(0编辑  收藏  举报