最新训练模型需要分别把数据分成训练集和测试集,用到了这个,记录下
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)
最新训练模型需要分别把数据分成训练集和测试集,用到了这个,记录下
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)