diff --git a/python-package/lightgbm/sklearn.py b/python-package/lightgbm/sklearn.py index 1195f342ecb..c31f2ff5b64 100644 --- a/python-package/lightgbm/sklearn.py +++ b/python-package/lightgbm/sklearn.py @@ -290,6 +290,10 @@ def __init__(self, boosting_type='gbdt', num_leaves=31, max_depth=-1, self._n_classes = None self.set_params(**kwargs) + def _more_tags(self): + return {'allow_nan': True, + 'X_types': ['2darray', 'sparse', '1dlabels']} + def get_params(self, deep=True): """Get parameters for this estimator.