CLASSIFIER_OUTPUT
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
Configs
Descriptions
inputNames
Optional. If set: this represents the NDArray. If not set: use DataUtils.inferField to find an NDArray field. Default is
null.returnLabel
Default is true; if false, don't return label.
returnIndex
Default is true; if false, don't return index.
returnProb
Default is true; if false, don't return probability.
labelName
Output names for the labels. Default value is
label.indexName
Output names for the index. Default value is
index.probName
Output names for the labels probabilities. Default value is
prob.labels
As a List<String>. Optional. If not specified, the predicted class index as a string is used - i.e.,
\"0\", \"1\", etc. Default value is null.topN
Integer,
null by default. If non-null and > 1, we return List<String>, List<Long>, List<Double> for the predicted class/index/probability instead of String/Long/Double.allProbabilities
If
true, also returns a List<List<Double>> of all probabilities (basically, convert NDArray to list). false by default.Last modified 1yr ago
Copy link