predict

fun predict(dataset: Dataset, batchSize: Int, callback: Callback): IntArray
Content copied to clipboard

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.

callback

Callback to be used during prediction phase.

abstract fun predict(dataset: Dataset, batchSize: Int, callbacks: List<Callback> = listOf()): IntArray
Content copied to clipboard

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.

callbacks

Callbacks to be used during prediction phase.

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.