关于train_test_split函数
train_test_split函数是交叉验证中常用的函数,可以按比例将数据划分成训练集和测试集。
导包
from sklearn.model_selection import train_test_split
使用
X_train, X_test, y_train, y_test = train_test_split(train_data,train_target,test_size, random_state=0,stratify)
参数解释
train_data指定训练集
train_target表示划分的样本结果
test_size:样本占比,如果是整数的话就是样本的数量
stratify指定训练集不同类别间的分配比例(处理不平衡数据)
random_state:是随机数的种子