Skip to content

Instantly share code, notes, and snippets.

@Tathagatd96
Last active July 28, 2019 03:13
Show Gist options
  • Save Tathagatd96/d1cc82ebec09be79a46f5eaf1f18b6a9 to your computer and use it in GitHub Desktop.
Save Tathagatd96/d1cc82ebec09be79a46f5eaf1f18b6a9 to your computer and use it in GitHub Desktop.
X_data=data.drop(["Class","Group"],axis=1)
y_data=data["Class"]
dtrain = xgb.DMatrix(X_data,y_data)
params = {
'objective':'binary:logistic',
'max-depth':2,
'silent':1,
'eta':0.5
}
num_rounds = 5
bst = xgb.train(params,dtrain,num_rounds)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment