diff --git a/Code/Day 1_Data_Preprocessing.md b/Code/Day 1_Data_Preprocessing.md index 5542709..d4d200c 100644 --- a/Code/Day 1_Data_Preprocessing.md +++ b/Code/Day 1_Data_Preprocessing.md @@ -41,7 +41,7 @@ Y = labelencoder_Y.fit_transform(Y) ``` ## 第5步:拆分数据集为训练集合和测试集合 ```python -from sklearn.cross_validation import train_test_split +from sklearn.model_selection import train_test_split X_train, X_test, Y_train, Y_test = train_test_split( X , Y , test_size = 0.2, random_state = 0) ``` ## 第6步:特征量化