added estimator's tags (#2150)

microsoft · May 6, 2019 · a47782f · a47782f
1 parent cfcc020
commit a47782f
Showing 1 changed file with 4 additions and 0 deletions.
diff --git 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.