Perdón,

preds <- colnames(probs)[apply(probs, 1, which.max)]

El 27/01/2021 a las 7:57, Marcelino de la Cruz Rot escribió:

preds <- col.names(probs)[apply(probs, 1, which.max)]



El 27/01/2021 a las 4:27, Manuel Mendoza escribió:
Buenos días, de un gbm multinomial obtengo los resultados en
probabilidades, como veis abajo.
Con   preds <- apply(probs, 1, which.max) obtengo la categoría más
probable, pero me la da como 1,1,1,3,2,5,... indicando la posición de la
categoría en vez de su nombre. Con ifelse se puede transformar fácilmente
pero, quizás, haya una forma de obtenerlo directamente. ¿Lo sabéis?

            CYT            ERL EXC                 ME1
          ME2             ME3
[1,] 0.4109855 0.0003585678 0.008226155 0.0010024805 0.011564794 0.011061570 [2,] 0.4481929 0.0003365305 0.020082853 0.0053111587 0.010587039 0.016284726 [3,] 0.5991016 0.0002893456 0.004667779 0.0007938136 0.010317520 0.004139039 [4,] 0.6953976 0.0001788419 0.002542308 0.0004987632 0.006486960 0.002685069 [5,] 0.1880506 0.0001065788 0.002126477 0.0002694346 0.005070764 0.005504816 [6,] 0.3271385 0.0001400425 0.002667148 0.0003842037 0.004645430 0.007562207

Gracias,
Manuel

    [[alternative HTML version deleted]]

_______________________________________________
R-help-es mailing list
R-help-es@r-project.org
https://stat.ethz.ch/mailman/listinfo/r-help-es
.



--
Marcelino de la Cruz Rot
Depto. de Biología y Geología
Física y Química Inorgánica
Universidad Rey Juan Carlos
Móstoles España

_______________________________________________
R-help-es mailing list
R-help-es@r-project.org
https://stat.ethz.ch/mailman/listinfo/r-help-es

Responder a