ClassDecoder[probability]

decodes a probability vector of 1 and 0 into Integers of 1 to the value of the last dimension of probability.

ClassDecoder[probability,nClass]

decodes a probability vector of 1 and 0 into Integers of 1 to nClass.