X_train, X_test, y_train, y_test = train_test_split(independent_variables, target, test_size=0.2, random_state=0)
时间: 2023-10-26 18:43:31 浏览: 114
数据集分割train和test程序
这是一个用于机器学习的代码示例,其中 independent_variables 是自变量的数据集,target 是因变量的数据集。train_test_split 函数将数据集划分为训练集和测试集,其中训练集占 80%,测试集占 20%。random_state 参数用于设置随机种子,以确保划分结果的可重复性。划分后的数据集分别存储在 X_train、X_test、y_train、y_test 四个变量中。
阅读全文