KNN-klassificering
I den här övningen utforskar du en delmängd av Large Movie Review Dataset.
Variablerna X_train, X_test, y_train och y_test är redan inlästa i miljön. Variablerna X innehåller särdrag baserade på orden i filmrecensionerna, och variablerna y innehåller etiketter för om recensionens sentiment är positivt (+1) eller negativt (-1).
Den här kursen berör många koncept som du kanske har glömt bort, så om du behöver en snabb repetition kan du ladda ned scikit-learn-facit och ha det till hands!
Den här övningen är en del av kursen
Linjära klassificerare i Python
Övningsinstruktioner
- Skapa en KNN-modell med standardhyperparametrar.
- Träna modellen.
- Skriv ut prediktionen för testexempel 0.
Interaktiv övning med praktiskt arbete
Testa den här övningen genom att slutföra den här exempelkoden.
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)