predictSoftly

abstract fun predictSoftly(dataset: Dataset, batchSize: Int): Array<FloatArray>

Generates output predictions for the input samples. Each prediction is a vector of probabilities instead of specific class in predict method.

Return

Array of labels. All labels are vectors that represents the probability distributions of a list of potential outcomes. The length is equal to the Number of samples on the dataset.

Parameters

dataset

Data to predict on.

batchSize

Number of samples per batch of computation.