Skip to content

GitLab

Unverified Commit 84c657e8 authored by James Lamb's avatar James Lamb Committed by GitHub
Browse files

[python-package] avoid data_has_header check in predict() (#5970)

parent 28e0e6f0
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 2 deletions
python-package/lightgbm/basic.py
View file @ 84c657e8
...@@ -961,14 +961,13 @@ class _InnerPredictor: ...@@ -961,14 +961,13 @@ class _InnerPredictor:
predict_type = _C_API_PREDICT_LEAF_INDEX predict_type = _C_API_PREDICT_LEAF_INDEX
if pred_contrib: if pred_contrib:
predict_type = _C_API_PREDICT_CONTRIB predict_type = _C_API_PREDICT_CONTRIB
int_data_has_header = 1 if data_has_header else 0
if isinstance(data, (str, Path)): if isinstance(data, (str, Path)):
with _TempFile() as f: with _TempFile() as f:
_safe_call(_LIB.LGBM_BoosterPredictForFile( _safe_call(_LIB.LGBM_BoosterPredictForFile(
self._handle, self._handle,
_c_str(str(data)), _c_str(str(data)),
ctypes.c_int(int_data_has_header), ctypes.c_int(data_has_header),
ctypes.c_int(predict_type), ctypes.c_int(predict_type),
ctypes.c_int(start_iteration), ctypes.c_int(start_iteration),
ctypes.c_int(num_iteration), ctypes.c_int(num_iteration),
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment