13

ランダム フォレスト分類を使用して R でモデルを構築しようとしています。(Ned Horning によるコードの編集による) 最初はrandomForestpackage を使用しましたが、その後ranger、より高速な計算を約束する を見つけました。

最初に、以下のコードを使用して、モデルを次のようにフィッティングした後、各クラスの予測確率を取得しrandomForestました。

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

ここでの確率のタイプは次のとおりです。

モデルには 500 本の木があり、そのうち 250 本は観察がクラス 1 であると言っているため、確率は 250/500 = 50% です。

では、選択肢rangerがないことに気づきました。type = 'prob'

検索して調整を試みましたが、進歩がありませんでした。上記のように、rangerパッケージで確率を含むオブジェクトなどが必要です。

この問題について誰かアドバイスをいただけますか?

4

1 に答える 1

14

「確率的分類子」タイプのrangerオブジェクトをトレーニングする必要があります。

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

predict.rangerこのオブジェクトは、関数で使用されると行列 (n_samples、n_classes) を計算します。

probabilities = predict(iris.ranger, data = iris)$predictions
于 2019-04-13T06:34:38.113 に答える