getMaxIndexAndProb function

(int, double) getMaxIndexAndProb(
  1. List<double> probabilities
)

Obtém o índice e a probabilidade máxima de uma lista de probabilidades.

Params:

  • probabilities: Lista de probabilidades para cada classe.

Returns: Tupla contendo o índice da classe com a maior probabilidade e o valor dessa probabilidade.

Implementation

(int, double) getMaxIndexAndProb(List<double> probabilities) {
  double maxProb = 0.0;
  int maxIndex = -1;
  for (int i = 0; i < probabilities.length; i++) {
    if (probabilities[i] > maxProb) {
      maxProb = probabilities[i];
      maxIndex = i;
    }
  }
  if (maxIndex == -1) {
    throw Exception('No classification result found.');
  }

  return (maxIndex, maxProb);
}