Skip to content

Instantly share code, notes, and snippets.

@okdolly-001

okdolly-001/xgb.sklearn.py

Last active March 16, 2018 08:53
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save okdolly-001/aed04337db8d9921f5cb4d3b39398a79 to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save okdolly-001/aed04337db8d9921f5cb4d3b39398a79 to your computer and use it in GitHub Desktop.
Download ZIP
xgb.sklearn
Raw
xgb.sklearn.py
clf = xgb.sklearn.XGBClassifier(
objective="binary:logistic",
learning_rate=0.05,
seed=9616,
max_depth=20,
gamma=10,
n_estimators=500)
clf.fit(X_train, Y_train, early_stopping_rounds=20, eval_metric="auc", eval_set=eval_set, verbose=True)
y_pred = clf.predict(X_test)
submission_file_name = 'Submission_'
print(y_pred)
https://github.com/avannaldas/Loan-Defaulter-Prediction-Machine-Learning/blob/master/Bank%20Fears%20Loanliness%20-%20Hacker%20Earth%20Machine%20Learning%20Challenge%201.ipynb
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment