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

问题描述

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

model_rf predict_p_rf

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

解决方法

使用 terra::predict 时更流畅,但使用 raster::predict 您可以使用 index 参数来指定您想要的输出变量。

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

?raster::predict

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