惰性训练-测试集划分
您已经完成了对 X 变量的变换。现在需要通过变换 y 变量,并将数据划分为训练集和测试集,来完成数据准备。
上一练习中您创建的变量 X 和 y 已加载到您的环境中。
本练习是课程的一部分
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)