From 8ca35c677f7512e735929d8ff22243e377e04ba8 Mon Sep 17 00:00:00 2001 From: Jack Huang Date: Tue, 21 Aug 2018 14:26:39 +0800 Subject: [PATCH] add R2 evaluation for linear_regression it will help to know the regression presicison --- Code/Day 3_Multiple_Linear_Regression.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/Code/Day 3_Multiple_Linear_Regression.py b/Code/Day 3_Multiple_Linear_Regression.py index ed54ae1..be6c4ec 100644 --- a/Code/Day 3_Multiple_Linear_Regression.py +++ b/Code/Day 3_Multiple_Linear_Regression.py @@ -29,3 +29,6 @@ regressor.fit(X_train, Y_train) # Predicting the Test set results y_pred = regressor.predict(X_test) +# regression evaluation +from sklearn.metrics import r2_score +print(r2_score(Y_test,y_pred))