Ich verwende die multinomiale Verteilung aus dem gbm -Paket in R. Wenn ich die Funktion predict verwende, bekomme ich eine Reihe von Werten:
aber ich möchte die Wahrscheinlichkeit für jede Klasse erhalten. Wie stelle ich die Wahrscheinlichkeiten wieder her? Danke.
predict.gbm(..., type='response') ist nicht für multinomiale oder andere Distributionen als bernoulli oder poisson implementiert.
Sie müssen also die wahrscheinlichste Klasse ( apply(.., 1, which.max) auf der Vektorausgabe von der Vorhersage) finden, wie Desertnaut schrieb :
Schreiben Sie einfach einen Wrapper, der type = 'response' akzeptiert und gibt diesen zurück, wenn es sich um ein multinomiales Modell handelt.
Tags und Links r prediction categorical-data multinomial gbm