Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Ensure pandas DataFrame column names are treated as strings in type error message (#4481)

Browse Source
This commit is contained in:
Sean Owen 2019-05-21 03:19:35 -05:00 committed by Jiaming Yuan
parent 515f5f5c47
commit 5a567ec249
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
python-package/xgboost/core.py
View File

@ -231,7 +231,7 @@ def _maybe_pandas_data(data, feature_names, feature_types):
data_dtypes = data.dtypes data_dtypes = data.dtypes
if not all(dtype.name in PANDAS_DTYPE_MAPPER for dtype in data_dtypes): if not all(dtype.name in PANDAS_DTYPE_MAPPER for dtype in data_dtypes):
bad_fields = [data.columns[i] for i, dtype in bad_fields = [str(data.columns[i]) for i, dtype in
enumerate(data_dtypes) if dtype.name not in PANDAS_DTYPE_MAPPER] enumerate(data_dtypes) if dtype.name not in PANDAS_DTYPE_MAPPER]
msg = """DataFrame.dtypes for data must be int, float or bool. msg = """DataFrame.dtypes for data must be int, float or bool.