Merge pull request #47 from Huang-Jack/patch-2

add R2 evaluation for linear_regression
This commit is contained in:
zhang yongquan
2018-08-21 14:40:07 +08:00
committed by GitHub

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