关于R中rpart构建CART模型的预测概率与类别代码验证问询

数据老陈

2026-6-20

关于rpart预测结果的确认

对于predictions：
你当前代码里，predict(..., type="prob")返回的是一个3列矩阵，每列对应类别1、2、3的预测概率，每行对应一个观测。但直接把这个矩阵赋值给iris$predictions会让它变成数据框里的列表列（每个单元格是包含3个概率的向量），并非只存储类别1的概率。如果只想保留类别1的概率，应该修改为：
```
iris$pred_prob_class1 <- predict(my_model, newdata = iris, type = "prob")[, 1]
```
对于pred_class：
完全正确，predict(..., type="class")返回的就是每个观测在三个类别中预测概率最大的那个类别，和你的理解一致。