数据集划分小探

数据集划分小探

x = [[0], [1], [2], [3],[4], [5], [6], [7],[8]]
y = [0, 0, 1, 1,2,2,2,2,3]

# # 2、数据基本处理 -- 划分数据集
x_train, x_test, y_train, y_test = sklearn.model_selection.train_test_split(x, y,test_size=0.2,random_state=1)
print(x_train)
print(x_test)
print(y_train)
print(y_test)

test_size=0.2
测试集或者训练集大小缺省时默认0.3
random_state=1
随机数种子缺省时训练数据集和测试数据集具体数字随机变动，加上可以调用相同的随机数种子方案，可以任意值