train_test_split 分割测试集训练集

Posted 2021-02-07 chengziaichiyu

#导入模块包
from sklearn.model_selection import train_test_split
 
#分割数据, test_size 默认是0.75
x_train, x_test, y_train, y_test = train_test_split(x, y, test_size=0.7,random_state=1)

test_size 设置测试集的比例；

random_state 固定随机种子，便于多次测试时使用。