andre/100-Days-Of-ML-Code
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add R2 evaluation for linear_regression

Browse Source 
it will help to know the regression presicison
This commit is contained in:
Jack Huang
2018-08-21 14:26:39 +08:00
committed by GitHub
parent 957fcc6853
commit 8ca35c677f
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
Code/Day 3_Multiple_Linear_Regression.py
View File

@ -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))