Add better error message for invalid feature names (#5024)

This commit is contained in:
mitama 2019-11-10 02:58:14 +09:00 committed by Jiaming Yuan
parent 7663de956c
commit 374648c21a

View File

@ -999,7 +999,7 @@ class DMatrix(object):
if not all(isinstance(f, STRING_TYPES) and
not any(x in f for x in set(('[', ']', '<')))
for f in feature_names):
raise ValueError('feature_names may not contain [, ] or <')
raise ValueError('feature_names must be string, and may not contain [, ] or <')
else:
# reset feature_types also
self.feature_types = None