Skip to content

Instantly share code, notes, and snippets.

@caiobvilar

caiobvilar/evaluating_model.py

Created December 3, 2019 14:12
Show Gist options
  • Save caiobvilar/1581720d41a7850d3ee171b5aeb8bc0f to your computer and use it in GitHub Desktop.
Save caiobvilar/1581720d41a7850d3ee171b5aeb8bc0f to your computer and use it in GitHub Desktop.
Download ZIP
Raw
evaluating_model.py
cv_params = {"objective":"reg:squarederror", "max_depth":4}
cv_results = xgb.cv(dtrain=DM_train, params=cv_params,nfold=10,num_boost_round=5, metrics="rmse",as_pandas=True,seed=42)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment