predict
Generates output predictions for the input samples.
Return
Array of labels. 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.
abstract fun predict(inputData: FloatArray, predictionTensorName: String): Int
Content copied to clipboard
Generates output prediction for the input sample using output of the predictionTensorName tensor.
Parameters
inputData
Unlabeled input data to define label.
predictionTensorName
Name of output tensor to make prediction.