Skip to content

Instantly share code, notes, and snippets.

@risenW
Created July 10, 2019 12:01
Show Gist options
  • Save risenW/79f99047ca73750a66e1e09b6d63b836 to your computer and use it in GitHub Desktop.
Save risenW/79f99047ca73750a66e1e09b6d63b836 to your computer and use it in GitHub Desktop.
#fit base models
linear_reg.fit(X_train, y_train)
knn_reg.fit(X_train, y_train)
svr_reg.fit(X_train, y_train)
#make predictions with trained models
pred1 = linear_reg.predict(X_val)
pred2 = knn_reg.predict(X_val)
pred3 = svr_reg.predict(X_val)
#Take average as final prediction
w_avgpred = (0.5 * pred1 + 0.25 * pred2 + 0.25* pred3)
print("Linear Regression Model")
print(get_mae(pred1, y_val))
print("KNN Regression Model")
print(get_mae(pred2, y_val))
print("SVR Regression Model")
print(get_mae(pred3, y_val))
print("Weighted Average Model")
print(get_mae(w_avgpred, y_val))
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment