MulaiMulai sekarang secara gratis

Pembagian train/test + menghitung akurasi

Sekarang saatnya berlatih membagi data Anda menjadi himpunan pelatihan dan pengujian dengan himpunan data churn_df!

Array NumPy telah disiapkan untuk Anda yang berisi fitur sebagai X dan variabel target sebagai y.

Latihan ini adalah bagian dari kursus

Supervised Learning dengan scikit-learn

Lihat Kursus

Petunjuk latihan

  • Impor train_test_split dari sklearn.model_selection.
  • Bagi X dan y menjadi himpunan pelatihan dan pengujian, tetapkan test_size sebesar 20%, random_state ke 42, dan pastikan proporsi label target mencerminkan himpunan data asli.
  • Latih model knn pada data pelatihan.
  • Hitung dan cetak akurasi model untuk data pengujian.

Latihan interaktif praktis

Cobalah latihan ini dengan menyelesaikan kode contoh berikut.

# Import the module
from ____ import ____

X = churn_df.drop("churn", axis=1).values
y = churn_df["churn"].values

# Split into training and test sets
X_train, X_test, y_train, y_test = ____(____, ____, test_size=____, random_state=____, stratify=____)
knn = KNeighborsClassifier(n_neighbors=5)

# Fit the classifier to the training data
____

# Print the accuracy
print(knn.score(____, ____))
Edit dan Jalankan Kode