train_test_split参数解释
创建时间:2021.10.20
在训练时我们时常需要划分数据集,这时常常使用sklearn.model_selection.train_test_split函数
常用方法:
X_train,X_test, y_train, y_test =sklearn.model_selection.train_test_split(train_data,train_target,test_size=0.4, random_state=0,stratify=y_train)
# train_data:所要划分的样本特征集
原创
2021-10-20 11:49:24 ·
4934 阅读 ·
0 评论