Skip to content

Instantly share code, notes, and snippets.

@vivekpadia70

vivekpadia70/house_prediction.py

Created December 25, 2019 06:30
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 vivekpadia70/aeeb6ff7aa606296febbc98c6dd1a9b0 to your computer and use it in GitHub Desktop.
Save vivekpadia70/aeeb6ff7aa606296febbc98c6dd1a9b0 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
house_prediction.py
params = {'n_estimators':[500, 1000, 1500, 2000], 'max_depth':[3, 5, 8]}
gbr = GradientBoostingRegressor()
gbr_grid = GridSearchCV(gbr, params, cv=5)
gbr_grid.fit(X_train, Y_train)
print("Grid Search Gradient Boosting Score: ", gbr_grid.score(X_train, Y_train))
print("Grid Search Gradient Boosting Test Score: ", gbr_grid.score(X_test, Y_test))
print("Grid Search Gradient Boosting Best Parameters: ", gbr_grid.best_params_)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment