caret::train 包中类型 = 'prob' 参数的问题

Problem with type = 'prob' argument in caret::train package

我正在训练一个随机森林模型,以使用 caret 基于 3 个分类变量 class 化光栅图像。我有兴趣获得这三个类别的 class 成员概率。这可以使用 type = 'prob' 来完成,但问题是我得到的只是一个单一的概率图像,范围从 0 到 1。

model_rf <- caret::train(xVar ~ . , method = "rf", data = dt_train, importance = TRUE, type="prob")
predict_p_rf <- raster::predict(object = image.x, model = model_rf, type = 'prob')

我的问题,1. 有没有办法在三个单独的输出中获得这三个类别的 class 成员概率? 2. 概率图像代表什么,因为它的范围从 0 到 1,同时存在三个不同的类别。我不确定更高的值(如 1)是否代表现有 classes/categories.

中任何一个的更高成员资格

这与 terra::predict 一起工作更顺畅,但与 raster::predict 一起,您可以使用 index 参数来指定您想要的输出变量。

predict_p_rf <- predict(image.x, model_rf, type = 'prob', index=1:3)

?raster::predict

数据表示属于特定类别的预测概率(0 是最低概率,1 是最高概率)。