isValidPrediction

private fun isValidPrediction(output: FloatArray): Boolean

Determina si los valores de salida del modelo TFLite indican una predicción válida.

Una predicción válida debe cumplir con los siguientes criterios:

  1. La salida debe contener exactamente dos valores flotantes.

  2. Cada valor debe representar una probabilidad y, por lo tanto, estar en el rango 0, 1.

  3. La suma de los dos valores debe ser aproximadamente 1 (con una tolerancia para errores de redondeo).

Return

true si la predicción es válida según los criterios, false en caso contrario.

Parameters

output

El array de flotantes que representa la salida del modelo TFLite. Se espera que contenga dos valores: P(no estrés), P(estrés).