Update 2-Regression/3-Linear/README.md

Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>
pull/967/head
Lee Stott 3 months ago committed by GitHub
parent cbcc13da01
commit c1f7e23ca8
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

@ -344,8 +344,8 @@ pipeline.fit(X_train,y_train)
pred = pipeline.predict(X_test)
# calculate RMSE and determination
rmse = np.sqrt(mean_squared_error(y_test,pred))
print(f'RMSE: {rmse:3.3} ({rmse/np.mean(pred)*100:3.3}%)')
rmse = mean_squared_error(y_test, pred, squared=False)
print(f'RMSE: {rmse:3.3} ({rmse/pred.mean()*100:3.3}%)')
score = pipeline.score(X_train,y_train)
print('Model determination: ', score)

Loading…
Cancel
Save