Commit e2a3b125 authored by Nikita Titov's avatar Nikita Titov Committed by Guolin Ke
Browse files

added check for pandas DataFrame dimentions (#1402)

parent af401561
...@@ -239,6 +239,8 @@ PANDAS_DTYPE_MAPPER = {'int8': 'int', 'int16': 'int', 'int32': 'int', ...@@ -239,6 +239,8 @@ PANDAS_DTYPE_MAPPER = {'int8': 'int', 'int16': 'int', 'int32': 'int',
def _data_from_pandas(data, feature_name, categorical_feature, pandas_categorical): def _data_from_pandas(data, feature_name, categorical_feature, pandas_categorical):
if isinstance(data, DataFrame): if isinstance(data, DataFrame):
if len(data.shape) != 2 or data.shape[0] < 1:
raise ValueError('Input data must be 2 dimensional and non empty.')
if feature_name == 'auto' or feature_name is None: if feature_name == 'auto' or feature_name is None:
data = data.rename(columns=str) data = data.rename(columns=str)
cat_cols = data.select_dtypes(include=['category']).columns cat_cols = data.select_dtypes(include=['category']).columns
......
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