Skip to content

Instantly share code, notes, and snippets.

@StrikingLoo
Last active April 13, 2019 01:19
Show Gist options
  • Save StrikingLoo/48340219de5d80ef55e19bfcde319e17 to your computer and use it in GitHub Desktop.
Save StrikingLoo/48340219de5d80ef55e19bfcde319e17 to your computer and use it in GitHub Desktop.
param = {'max_depth':5,
'eta':.1,
'colsample_bytree':.75
}
num_round = 40
new_tree = xgb.train(param, dtrain, num_round)
# make prediction
new_preds = new_tree.predict(dtest)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment