小编Fab*_*ana的帖子

scikit-learn 中 predict 与 predict_proba 的区别

假设我创建了一个模型,我的目标变量是0,12。似乎如果我使用predict,答案是 0、1 或 2。但是如果我使用predict_proba,我会得到一行,每行有 3 个列,如下所示,例如

   model = ... Classifier       # It could be any classifier
   m1 = model.predict(mytest)
   m2= model.predict_proba(mytest)

   # Now suppose  m1[3] = [0.6, 0.2, 0.2]
Run Code Online (Sandbox Code Playgroud)

假设我同时使用 predict 和predict_proba. 如果在索引 3 中,我得到了上面的结果predict_proba,在 predict 结果的索引 3 中,我应该看到 0。是这种情况吗?我试图了解如何在同一模型上使用predict和使用两者predict_proba相互关联。

python classification machine-learning prediction scikit-learn

6
推荐指数
1
解决办法
4951
查看次数