added estimator's tags (#2150)

a47782f5 · Nikita Titov · Guolin Ke · cfcc020e · a47782f5
Commit a47782f5 authored May 06, 2019 by Nikita Titov Committed by Guolin Ke May 06, 2019
Show whitespace changes
Inline Side-by-side

Showing with 4 additions and 0 deletions

python-package/lightgbm/sklearn.py python-package/lightgbm/sklearn.py +4 -0

No files found.
--- a/python-package/lightgbm/sklearn.py
+++ b/python-package/lightgbm/sklearn.py
@@ -290,6 +290,10 @@ class LGBMModel(_LGBMModelBase):
        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.