ClassifierOutputStep takes as input a numerical 2d NDArray (i.e., float/double etc. type) with shape [minibatch, numClasses] which represents the softmax predictions for a standard classifier and returns based on this array:

  1. The predicted class label - as a String

  2. The predicted class index - as an integer (long)

  3. The predicted class probability - as a Double

Last updated