El método predict

La implementación del método predict es, con lo que ya hemos visto, bastante sencilla: no tenemos más que considerar cada una de las muestras para las que haya que calcular la predicción y hacerla pasar por la red para obtener el resultado buscado. Los únicos matices a considerar son:

  • Invocaremos este método con un dataframe Pandas conteniendo las muestras para las que realizar la predicción, lo que nos obligará a extraer cada muestra y a redimensionarla adecuadamente.
  • Estamos implementando un clasificador, por lo que recordemos que la salida de la red para una muestra será un vector con tantos valores como posibles categorías estemos considerando. La predicción será el índice del valor más elevado.