【发布时间】:2017-08-13 16:09:33
【问题描述】:
我正在使用rpart 包来制作预测模型。如果我使用predict 函数并设置type="class",它会根据概率自动生成类。
我想知道这是基于的截止概率值,但似乎无法在任何地方找到它。
【问题讨论】:
标签: r class probability rpart
我正在使用rpart 包来制作预测模型。如果我使用predict 函数并设置type="class",它会根据概率自动生成类。
我想知道这是基于的截止概率值,但似乎无法在任何地方找到它。
【问题讨论】:
标签: r class probability rpart
它根据概率值分配类别。如果它在预测 1 和 0 之间,则默认截止值为 0.5,同样,它会根据响应变量的类别数量而有所不同。
【讨论】:
rpart 基于 CART,该算法在“Breiman L., Friedman J. H., Olshen R. A., and Stone, C. J. (1984) Classification and Regression Trees”中提出。它不使用概率来决定如何拆分节点,因此没有这样的阈值。相反,CART 使用节点的杂质度量。 rpart 使用 Gini(默认值)或信息增益来测量杂质。 有您可以使用rpart.control 设置一些阈值。三个主要是cpminsplit 和minbucket,用rpart.control 描述。
Breiman 的书是这方面的主要参考资料,但可以从 Wikipedia
【讨论】: