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 @   Right2014  阅读(2080)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· TypeScript + Deepseek 打造卜卦网站:技术与玄学的结合
· 阿里巴巴 QwQ-32B真的超越了 DeepSeek R-1吗?
· 【译】Visual Studio 中新的强大生产力特性
· 2025年我用 Compose 写了一个 Todo App
· 张高兴的大模型开发实战:(一)使用 Selenium 进行网页爬虫
点击右上角即可分享
微信分享提示