Поділ на train/test і обчислення точності
Час попрактикуватися в поділі даних на тренувальну й тестову вибірки з набором churn_df!
Для вас вже створено масиви NumPy: ознаки в X, а цільова змінна в y.
Ця вправа є частиною курсу
Кероване навчання зі scikit-learn
Інструкції до вправи
- Імпортуйте
train_test_splitзsklearn.model_selection. - Розбийте
Xіyна тренувальну та тестову вибірки, встановившиtest_sizeрівним 20%,random_state—42, і забезпечивши, щоб пропорції цільових міток відповідали оригінальному набору даних. - Навчіть модель
knnна тренувальних даних. - Обчисліть і виведіть точність моделі на тестових даних.
Інтерактивна практична вправа
Спробуйте виконати цю вправу, доповнивши цей зразок коду.
# 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(____, ____))