决策树的predict_pro

2020-03-04  本文已影响0人  番茄酱的汪

我们都知道决策树是以信息增益大小来确定分类的属性的,然后形成一棵树,那到了最后为什么还有predict_pro呢?
我理解的,应该是到了最后的叶子节点的时候,就直接根据样本量的大小推测概率大小,而不是直接一分为二。
所以预测的时候到这个叶子节点就是根据样本量的大小来推测概率值的。如果直接预测类别应该是以最大概率作为其最终概率。

是不是大家都有这个疑问呢?

上一篇 下一篇

猜你喜欢

热点阅读