Clasificación con KNN
En este ejercicio vas a explorar un subconjunto del Large Movie Review Dataset.
Las variables X_train, X_test, y_train y y_test ya están cargadas en el entorno. Las variables X contienen características basadas en las palabras de las reseñas de películas, y las variables y contienen etiquetas que indican si el sentimiento de la reseña es positivo (+1) o negativo (-1).
Este curso repasa muchos conceptos que quizá hayas olvidado; si necesitas un recordatorio rápido, descarga la scikit-learn Cheat Sheet y tenla a mano.
Este ejercicio forma parte del curso
Clasificadores lineales en Python
Instrucciones del ejercicio
- Crea un modelo KNN con los hiperparámetros predeterminados.
- Ajusta el modelo.
- Imprime la predicción para el ejemplo de prueba 0.
ejercicio interactivo práctico
Prueba este ejercicio completando este código de ejemplo.
from sklearn.neighbors import KNeighborsClassifier
# Create and fit the model
knn = ____
knn.____
# Predict on the test features, print the results
pred = knn.____[0]
print("Prediction for test example 0:", pred)