predict

open override fun predict(inputData: FloatArray): Int
Content copied to clipboard
fun predict(inputData: FloatArray, inputTensorName: String, outputTensorName: String): Int
Content copied to clipboard

Predicts the class of inputData.

Return

Predicted class index.

Parameters

inputData

The single example with unknown label.

inputTensorName

The name of input tensor.

outputTensorName

The name of output tensor.