1. Learn
    2. /
  2. 课程
    3. /
  3. 使用 scikit-learn 的监督学习
Connected

道练习

训练/测试集划分与计算准确率

现在用 churn_df 数据集练习将数据划分为训练集和测试集吧！

我们已为您创建了 NumPy 数组：特征为 X，目标变量为 y。

说明

100 XP
  • 从 sklearn.model_selection 导入 train_test_split。
  • 将 X 和 y 划分为训练集与测试集，设置 test_size 为 20%，random_state 为 42，并确保目标标签的比例与原始数据集一致。
  • 将 knn 模型拟合到训练数据。
  • 计算并打印模型在测试数据上的准确率。