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

Fix with latest panda range index. (#7074)

Browse Source
This commit is contained in:
Jiaming Yuan 2021-07-03 16:43:52 +08:00 committed by GitHub
parent 93f3acdef9
commit b56d3d5d5c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
python-package/xgboost/data.py
View File

@ -197,7 +197,7 @@ _pandas_dtype_mapper = {
def _transform_pandas_df(data, enable_categorical, def _transform_pandas_df(data, enable_categorical,
feature_names=None, feature_types=None, feature_names=None, feature_types=None,
meta=None, meta_type=None): meta=None, meta_type=None):
from pandas import MultiIndex, Int64Index from pandas import MultiIndex, Int64Index, RangeIndex
from pandas.api.types import is_sparse, is_categorical_dtype from pandas.api.types import is_sparse, is_categorical_dtype
data_dtypes = data.dtypes data_dtypes = data.dtypes
@ -219,7 +219,7 @@ def _transform_pandas_df(data, enable_categorical,
feature_names = [ feature_names = [
' '.join([str(x) for x in i]) for i in data.columns ' '.join([str(x) for x in i]) for i in data.columns
] ]
elif isinstance(data.columns, Int64Index): elif isinstance(data.columns, (Int64Index, RangeIndex)):
feature_names = list(map(str, data.columns)) feature_names = list(map(str, data.columns))
else: else:
feature_names = data.columns.format() feature_names = data.columns.format()