我正在嘗試使用r中的rpart包創建決策樹。爲了達到樹的最佳深度,我使用plotcp
函數。當我使用printcp
分析交叉驗證的結果,其他的細節中,我得到以下信息:修剪rpart樹
Root node error: 3599.8/14399 = 0.25
我的課是不平衡的(1-75級%,2-25級%)。那麼rpart似乎在做的是使用0.5的默認閾值。而且由於沒有一個節點對C2類的概率> 0.5,它們都被歸類爲C1。
我不可能指定概率閾值嗎?說,例如,如果C2的概率> 0.35,則將其歸類爲C2。