A ver qué tal esto (sin pensarlo mucho):
levels(categorÃas)[apply(probs, 1, which.max)]
El 19/03/2019 a las 13:07, Manuel Mendoza escribió:
Buenos dÃas erreros. Aplico un gbm con distribution multinomial, la
variable objetivo como factor, y obtengo las predicciones con
predict(), que me da las probabilidades de cada categorÃa, a lo que
aplico un apply(probs, 1, which.max) para obtener la predicción. El
problema está en que me la da con el número al que corresponde cada
categorÃa, en vez de con la categorÃa. Probé a poner la variable
objetivo, o el resultado del apply como character, pero sale igual.
Gracias,
Manuel
.