splitting = train_test_split(*arrays,**options)
如:
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.25, random_state=0, shuffle=False)
| 参数 | 参数说明 | 备注 | |
|---|---|---|---|
| splitting | X_train | list, length=2 * len(arrays) List, length=2 * len(数组) 包含输入的训练测试分割的列表。 |
划分后的训练数据集X轴 |
| X_test | 划分后的测试数据集X轴 | ||
| y_train | 划分后的训练数据集Y轴 | ||
| y_test | 划分后的测试数据集Y轴 | ||

本文介绍如何使用sklearn库中的train_test_split函数来划分数据集,并通过实例演示了如何进行数据集的划分及利用线性回归模型进行预测。此外还详细解释了各参数的作用,包括随机种子、分层划分等。

4706

被折叠的 条评论
为什么被折叠?



