Skip to content

Instantly share code, notes, and snippets.

@vikeshsingh37

vikeshsingh37/stack_model_example.py

Created February 23, 2019 16:17
Show Gist options
  • Save vikeshsingh37/4b6e79cb6e79f9f64eaf3da3676fa564 to your computer and use it in GitHub Desktop.
Save vikeshsingh37/4b6e79cb6e79f9f64eaf3da3676fa564 to your computer and use it in GitHub Desktop.
Download ZIP
Example to use stack_model.py
Raw
stack_model_example.py
from sklearn.ensemble import RandomForestRegressor, GradientBoostingRegressor
from sklearn.linear_model import ElasticNet
from stack_model import stack_model
rf_model = RandomForestRegressor()
gbm_model = GradientBoostingRegressor()
en_model = ElasticNet()
base_models = [rf_model,gbm_model,en_model]
meta_model = RandomForestRegressor(n_estimators =100)
stk_model = stack_model(base_models,meta_model,X[selected_features],y.values)
stk_model.fit(cvFolds = 5, random_state= 42)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment