使用train_test_split 分割测试集以及训练集

最新训练模型需要分别把数据分成训练集和测试集,用到了这个,记录下

from sklearn.model_selection import train_test_split    

X=df.iloc[:,0:4].values.tolist()
y=df.iloc[:,4]
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.33, random_state=42)

 

你可能感兴趣的:(Python)