创建两个留出集
您最近为老板构建了一个用于预测井字棋(Tic-Tac-Toe)胜负的简单随机森林模型,并且按她的要求没有进行参数调优。遗憾的是,模型的整体准确率达不到她的标准。这一次,她要求您专注于模型性能。
在开始测试不同的模型和参数组合之前,您需要将数据划分为训练集、验证集和测试集。请记住,在先将数据划分为训练集和测试集之后,验证集是通过再次划分训练集而创建的。
数据集 X 和 y 已为您加载。
本练习是课程的一部分
Python 中的模型验证
练习说明
- 创建临时数据集和测试数据集(
X_test、y_test)。测试集使用整体数据的 20%。 - 使用临时数据集(
X_temp、y_temp)创建训练集(X_train、y_train)和验证集(X_val、y_val)。 - 验证集使用「临时数据」的 25%。
交互式实操练习
通过完成这段示例代码来试试这个练习。
# Create temporary training and final testing datasets
X_temp, ____, y_temp, ____ =\
train_test_split(X, y, ____=____, random_state=1111)
# Create the final training and validation datasets
____, ____, ____, ____ =\
train_test_split(X_temp, y_temp, ____=____, random_state=1111)