Fix a comment in demo to use correct reference (#6190)

Co-authored-by: John Quitto Graham <johnq@dgx07.aselab.nvidia.com>
This commit is contained in:
John Quitto-Graham 2020-10-01 13:16:04 -07:00 committed by GitHub
parent eb7946ff25
commit e0e4f15d0e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -43,7 +43,7 @@ def evalerror(preds, dtrain):
py_evals_result = {} py_evals_result = {}
# training with customized objective, we can also do step by step training # training with customized objective, we can also do step by step training
# simply look at xgboost.py's implementation of train # simply look at training.py's implementation of train
py_params = param.copy() py_params = param.copy()
py_params.update({'disable_default_eval_metric': True}) py_params.update({'disable_default_eval_metric': True})
py_logreg = xgb.train(py_params, dtrain, num_round, watchlist, obj=logregobj, py_logreg = xgb.train(py_params, dtrain, num_round, watchlist, obj=logregobj,