diff --git a/Code/Day 3_Multiple_Linear_Regression.py b/Code/Day 3_Multiple_Linear_Regression.py index be6c4ec..291598f 100644 --- a/Code/Day 3_Multiple_Linear_Regression.py +++ b/Code/Day 3_Multiple_Linear_Regression.py @@ -18,7 +18,7 @@ X = onehotencoder.fit_transform(X).toarray() X = X[: , 1:] # Splitting the dataset into the Training set and Test set -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) # Fitting Multiple Linear Regression to the Training set