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)
作者: Right2014
出处:https://www.cnblogs.com/ALice1024/p/16314028.html
版权:本站使用「CC BY 4.0」创作共享协议,转载请在文章明显位置注明作者及出处。
Motto:Stay Hungry,Stay Foolish
ContactMe:rightsec1024@gmail.com
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· TypeScript + Deepseek 打造卜卦网站:技术与玄学的结合
· 阿里巴巴 QwQ-32B真的超越了 DeepSeek R-1吗?
· 【译】Visual Studio 中新的强大生产力特性
· 2025年我用 Compose 写了一个 Todo App
· 张高兴的大模型开发实战:(一)使用 Selenium 进行网页爬虫