diff --git a/Code/Day 1_Data_Preprocessing.py b/Code/Day 1_Data_Preprocessing.py index fedb09f..b2a6117 100644 --- a/Code/Day 1_Data_Preprocessing.py +++ b/Code/Day 1_Data_Preprocessing.py @@ -42,7 +42,7 @@ print("Y") print(Y) #Step 5: Splitting the datasets into training sets and Test sets -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) print("---------------------") print("Step 5: Splitting the datasets into training sets and Test sets")