始める無料で始める

KNNによる分類

この演習では、Large Movie Review Dataset の一部を扱います。 変数 X_trainX_testy_trainy_test はすでに環境に読み込まれています。X には映画レビューの単語に基づく特徴量が、y にはレビューの感情がポジティブ(+1)かネガティブ(-1)かのラベルが入っています。

このコースでは、忘れているかもしれない概念も多く登場します。必要になったら、scikit-learn チートシート をダウンロードして手元に置いておくと便利です。

この演習はコースの一部です

Python で学ぶ線形分類器

コースを見る

演習の手順

  • 既定のハイパーパラメータで KNN モデルを作成します。
  • モデルを学習させます(fit します)。
  • テストデータのサンプル 0 に対する予測を出力します。

実践的なインタラクティブ演習

このサンプルコードを完成させて、この演習に挑戦してみましょう。

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)
コードを編集して実行