Skip to content

Instantly share code, notes, and snippets.

@maskaravivek
Created February 13, 2021 06:21
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save maskaravivek/41ec126fc06c14681540e34151ab86b7 to your computer and use it in GitHub Desktop.
Save maskaravivek/41ec126fc06c14681540e34151ab86b7 to your computer and use it in GitHub Desktop.
from sklearn.ensemble import GradientBoostingRegressor
gb = GradientBoostingRegressor(loss ='huber', n_estimators=300, max_depth=5,)
gb.fit(X_train,y_train)
ypred = gb.predict(X_test)
plot_predictions(y_test, ypred)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment