Corrected the output to be instead the index of the max values of the one-hot-encoded prediction labels along the second axis.
Corrected the output to be instead the index of the max values of the one-hot-encoded prediction labels along the second axis.

MizioAnd/PortoSeguroInsur