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
Instrucciones de ejercicio
- Añade un título
"KNN: Varying Number of Neighbors"
. - Traza el método
.values()
detrain_accuracies
en el eje y contraneighbors
en el eje x, con una etiqueta de"Training Accuracy"
. - Traza el método
.values()
detest_accuracies
en el eje y contraneighbors
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
____