EmpezarEmpieza gratis

Predicciones con softmax

Tu model recién entrenado está cargado. ¡Este modelo está generalizando bien!, por eso obtuviste una alta accuracy en el conjunto de prueba.

Como usaste la función de activación softmax, para cada entrada de 2 coordenadas que das a tu modelo hay un vector de salida de 4 números. Cada uno de estos números codifica la probabilidad de que un dardo haya sido lanzado por uno de los 4 posibles competidores.

Al calcular la accuracy con el método .evaluate() del modelo, tu modelo toma como predicción la clase con mayor probabilidad. np.argmax() puede ayudarte con esto, ya que devuelve el índice con el valor más alto en un array.

¡Usa la colección de lanzamientos de prueba guardada en coords_small_test y np.argmax() para comprobarlo!

Este ejercicio forma parte del curso

Introducción al Deep Learning con Keras

Ver curso

ejercicio interactivo práctico

Prueba este ejercicio completando este código de ejemplo.

# Predict on coords_small_test
preds = ____.____(____)

# Print preds vs true values
print("{:45} | {}".format('Raw Model Predictions','True labels'))
for i,pred in enumerate(preds):
  print("{} | {}".format(pred,competitors_small_test[i]))
Editar y ejecutar código