ComenzarEmpieza gratis

Visualizar la complejidad del modelo

Ahora que has calculado la precisión del modelo KNN en los conjuntos de entrenamiento y prueba utilizando varios valores de n_neighbors, puedes crear una curva de complejidad del modelo para visualizar cómo cambia el rendimiento a medida que el modelo se hace menos complejo.

Se han precargado las variables neighbors, train_accuracies y test_accuracies, que generaste en el ejercicio anterior. Trazarás los resultados para ayudarte a encontrar el número óptimo de vecinos para tu modelo.

Este ejercicio forma parte del curso

Aprendizaje supervisado con scikit-learn

Ver curso

Instrucciones de ejercicio

  • Añade un título "KNN: Varying Number of Neighbors".
  • Traza el método .values() de train_accuracies en el eje y contra neighbors en el eje x, con una etiqueta de "Training Accuracy".
  • Traza el método .values() de test_accuracies en el eje y contra neighbors en el eje x, con una etiqueta de "Testing Accuracy".
  • Visualiza el gráfico.

Ejercicio interactivo práctico

Pruebe este ejercicio completando este código de muestra.

# Add a title
plt.title("____")

# Plot training accuracies
plt.plot(____, ____, label="____")

# Plot test accuracies
plt.plot(____, ____, label="____")

plt.legend()
plt.xlabel("Number of Neighbors")
plt.ylabel("Accuracy")

# Display the plot
____
Editar y ejecutar código