predictSoftly

abstract fun predictSoftly(inputData: FloatArray, predictionTensorName: String = ""): FloatArray

Predicts vector of probabilities instead of specific class in predict method.

Return

Vector that represents the probability distributions of a list of potential outcomes

Parameters

inputData

The single example with unknown vector of probabilities.

predictionTensorName

The name of prediction tensor. It could be changed, if you need to get alternative outputs from model graph.