diff --git a/Code/Day 1_Data PreProcessing.md b/Code/Day 1_Data PreProcessing.md index 569d0e6..9489031 100644 --- a/Code/Day 1_Data PreProcessing.md +++ b/Code/Day 1_Data PreProcessing.md @@ -39,7 +39,7 @@ Y = labelencoder_Y.fit_transform(Y) ``` ## Step 5: Splitting the datasets into training sets and Test sets ```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) ```