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

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)