Skip to content

Commit 977d3e7

Browse files
author
ShvetsKS
committed
apply comments
1 parent 99ab056 commit 977d3e7

File tree

2 files changed

+3
-2
lines changed

2 files changed

+3
-2
lines changed

configs/xgb_gpu_config.json

100755100644
Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -129,7 +129,8 @@
129129
"max-leaves": [256],
130130
"n-estimators": [1000],
131131
"objective": ["binary:logistic"],
132-
"tree-method": ["gpu_hist"]
132+
"tree-method": ["gpu_hist"],
133+
"inplace-predict": [""]
133134
},
134135
{
135136
"algorithm": "gbt",

xgboost/gbt.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -68,7 +68,7 @@ def convert_xgb_predictions(y_pred, objective):
6868
parser.add_argument('--count-dmatrix', default=False, action='store_true',
6969
help='Count DMatrix creation in time measurements')
7070
parser.add_argument('--inplace-predict', default=False, action='store_true',
71-
help='Count DMatrix creation in time measurements')
71+
help='Perform inplace_predict instead of default')
7272
parser.add_argument('--single-precision-histogram', default=False, action='store_true',
7373
help='Build histograms instead of double precision')
7474
parser.add_argument('--enable-experimental-json-serialization', default=True,

0 commit comments

Comments
 (0)