Skip to content

Instantly share code, notes, and snippets.

@joshreini1

joshreini1/airbnb_xgb_reg.py

Created August 30, 2022 19:47
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 joshreini1/e876655e217856a55077542b430c99f9 to your computer and use it in GitHub Desktop.
Save joshreini1/e876655e217856a55077542b430c99f9 to your computer and use it in GitHub Desktop.
Download ZIP
instantiate xgboost regression for airbnb
Raw
airbnb_xgb_reg.py
#instantiate model
xgb_reg = xgb.XGBRegressor()
# combine both preprocessing and modeling
xgb_pipe = Pipeline([
('preprocess', comprehensive_preprocessing),
('xgboost', xgb_reg)
])
#fit pipe
xgb_pipe.fit(X_train,y_train)
#add model to truera
tru.add_python_model("xgb_pipe", xgb_pipe, pipeline_feature_map = feature_map_data)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment