开始使用免费开始使用

惰性训练-测试集划分

您已经完成了对 X 变量的变换。现在需要通过变换 y 变量,并将数据划分为训练集和测试集,来完成数据准备。

上一练习中您创建的变量 Xy 已加载到您的环境中。

本练习是课程的一部分

Python 中的 Dask 并行编程

查看课程

练习说明

  • dask_ml.model_selection 导入 train_test_split() 函数。
  • y 中的人气分数范围为 0-100,将其除以 100,使其落在 0-1 范围内。
  • 使用 train_test_split() 将数据划分为训练集和测试集,确保打乱数据,并将测试集比例设置为数据的 20%。

交互式实操练习

通过完成这段示例代码来试试这个练习。

# Import the train_test_split function
from ____ import ____

# Rescale the target values
y = ____

# Split the data into train and test sets
X_train, X_test, y_train, y_test = ____

print(X_train)
编辑并运行代码