sklearn.model_selection中的train_test_split使用方法，分割数据集，划分训练集和测试集

随手笔记_000333

于 2017-12-28 16:50:55 发布

阅读量3k

点赞数 1

CC 4.0 BY-SA版权

分类专栏：深度学习

本文链接：https://blog.youkuaiyun.com/SuperYR_210/article/details/78924013

import numpy as np
from sklearn.model_selection import train_test_split


X = np.load("./data/111.npy")
y = np.load("./data/222.npy")


print(X.shape)
print(y.shape)

X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=0)          #test_size占20%

print(X_train.shape)