From d263be841edbf3a8fd485b119db8921125add07d Mon Sep 17 00:00:00 2001 From: wengJJ <31797645+wengJJ@users.noreply.github.com> Date: Thu, 6 Sep 2018 22:27:54 +0800 Subject: [PATCH] Update Day 3_Multiple_Linear_Regression.py --- Code/Day 3_Multiple_Linear_Regression.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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