KNN classificação

Neste exercício, você explorará um subconjunto do Large Movie Review Dataset. As variáveis X_train, X_test, y_train e y_test já estão carregadas no ambiente. As variáveis X contêm recursos baseados nas palavras das resenhas de filmes, e as variáveis y contêm rótulos que indicam se o sentimento da resenha é positivo (+1) ou negativo (-1).

Este curso aborda muitos conceitos que você pode ter esquecido, portanto, se precisar de uma atualização rápida, baixe a Folha de dicas do scikit-learn e mantenha-a sempre à mão!

Este exercício faz parte do curso

Classificadores lineares em Python

Ver Curso

Instruções de exercício

  • Crie um modelo KNN com hiperparâmetros padrão.
  • Ajuste o modelo.
  • Imprima a previsão para o exemplo de teste 0.

Exercício interativo prático

Experimente este exercício preenchendo este código de exemplo.

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)