问题描述
我正在训练一个随机森林模型,以使用基于 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 为最高)。