gpt4 book ai didi

r - R ranger 包中的预测概率

转载 作者:行者123 更新时间:2023-12-02 00:39:20 24 4
gpt4 key购买 nike

我正在尝试使用随机森林分类在 R 中构建模型。 (通过 Ned Horning 编辑代码)我首先使用了 randomForest 包,但后来发现了 ranger,它保证了更快的计算。

首先,在使用randomForest拟合模型后,我使用下面的代码来获取每个类的预测概率:

predProbs <- as.data.frame(predict(randfor, imageBlock, type='prob'))

这里的概率类型如下:

模型中有 500 棵树,其中 250 棵表示观察结果为 1 类,因此概率为 250/500 = 50%

ranger中,我意识到没有type = 'prob'选项。

我搜索并尝试了一些调整,但没有取得任何进展。我需要一个包含上述 ranger 包概率的对象。

有人可以就这个问题提供一些建议吗?

最佳答案

您需要训练一个“概率分类器”类型的ranger对象:

library("ranger")
iris.ranger = ranger(Species ~ ., data = iris, probability = TRUE)

此对象在 predict.ranger 函数中使用时计算矩阵(n_samples、n_classes):

probabilities = predict(iris.ranger, data = iris)$predictions

关于r - R ranger 包中的预测概率,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/55654644/

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com