Skip to content

GitLab

Commit e4dde71f authored by Guolin Ke's avatar Guolin Ke Committed by GitHub
Browse files

Update basic.py

parent 13226fa3
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
python-package/lightgbm/basic.py
View file @ e4dde71f
...@@ -535,8 +535,8 @@ class _InnerDataset(object): ...@@ -535,8 +535,8 @@ class _InnerDataset(object):
"""start construct data""" """start construct data"""
if is_str(data): if is_str(data):
"""check data has header or not""" """check data has header or not"""
if params.get("has_header", "").lower() == "true" \ if str(params.get("has_header", "")).lower() == "true" \
or params.get("header", "").lower() == "true": or str(params.get("header", "")).lower() == "true":
self.data_has_header = True self.data_has_header = True
self.handle = ctypes.c_void_p() self.handle = ctypes.c_void_p()
_safe_call(_LIB.LGBM_DatasetCreateFromFile( _safe_call(_LIB.LGBM_DatasetCreateFromFile(
......
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