Skip to content

Instantly share code, notes, and snippets.

@jkotra

jkotra/xgb.py

Created August 29, 2018 17:30
Show Gist options
  • Save jkotra/bf947c47965badfbf986e2d9192a1a9a to your computer and use it in GitHub Desktop.
Save jkotra/bf947c47965badfbf986e2d9192a1a9a to your computer and use it in GitHub Desktop.
Download ZIP
taxi-xgb
Raw
xgb.py
parms = {'max_depth': 8,'learning_rate':0.07, 'eta': 0.03, 'subsample': 1, 'colsample_bytree': 0.8, 'objective': 'reg:linear', 'eval_metric': 'rmse', 'silent': 0}
reg = xgb.XGBRegressor(parms=parms,n_jobs=6)
reg = xgb.train(parms, dtrain, num_boost_round=500)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment