code_4.py 135 字节
Newer Older
ToTensor's avatar
ToTensor 已提交
1 2
from sklearn.model_selection import train_test_split #导入sklearn工具库
X_train, X_test, y_train, y_test = train_test_split(X, y,