diff --git a/multiple_linear_regression.py b/multiple_linear_regression.py index 5cdc532..f2adf16 100644 --- a/multiple_linear_regression.py +++ b/multiple_linear_regression.py @@ -15,7 +15,7 @@ boston['MEDV'] = bh_data.target -X = pd.DataFrame(np.c_[boston['LSTAT'], boston['RM']], columns=['LSTAT','RM']) +X=boston[['LSTAT' , 'RM']] y = boston['MEDV'] X_train, X_test, y_train, y_test = train_test_split(X, y, test_size = 0.2, random_state=9)