Unverified Commit 56b99d4c authored by James Lamb's avatar James Lamb Committed by GitHub
Browse files

[dask] rename _LGBMModel to _DaskLGBMModel (#3859)

parent 9f70e968
...@@ -377,7 +377,7 @@ def _predict(model, data, raw_score=False, pred_proba=False, pred_leaf=False, pr ...@@ -377,7 +377,7 @@ def _predict(model, data, raw_score=False, pred_proba=False, pred_leaf=False, pr
raise TypeError('Data must be either Dask array or dataframe. Got %s.' % str(type(data))) raise TypeError('Data must be either Dask array or dataframe. Got %s.' % str(type(data)))
class _LGBMModel: class _DaskLGBMModel:
def __init__(self): def __init__(self):
if not all((DASK_INSTALLED, PANDAS_INSTALLED, SKLEARN_INSTALLED)): if not all((DASK_INSTALLED, PANDAS_INSTALLED, SKLEARN_INSTALLED)):
raise LightGBMError('dask, pandas and scikit-learn are required for lightgbm.dask') raise LightGBMError('dask, pandas and scikit-learn are required for lightgbm.dask')
...@@ -419,7 +419,7 @@ class _LGBMModel: ...@@ -419,7 +419,7 @@ class _LGBMModel:
setattr(dest, name, attributes[name]) setattr(dest, name, attributes[name])
class DaskLGBMClassifier(LGBMClassifier, _LGBMModel): class DaskLGBMClassifier(LGBMClassifier, _DaskLGBMModel):
"""Distributed version of lightgbm.LGBMClassifier.""" """Distributed version of lightgbm.LGBMClassifier."""
def fit(self, X, y=None, sample_weight=None, client=None, **kwargs): def fit(self, X, y=None, sample_weight=None, client=None, **kwargs):
...@@ -467,7 +467,7 @@ class DaskLGBMClassifier(LGBMClassifier, _LGBMModel): ...@@ -467,7 +467,7 @@ class DaskLGBMClassifier(LGBMClassifier, _LGBMModel):
return self._to_local(LGBMClassifier) return self._to_local(LGBMClassifier)
class DaskLGBMRegressor(LGBMRegressor, _LGBMModel): class DaskLGBMRegressor(LGBMRegressor, _DaskLGBMModel):
"""Docstring is inherited from the lightgbm.LGBMRegressor.""" """Docstring is inherited from the lightgbm.LGBMRegressor."""
def fit(self, X, y=None, sample_weight=None, client=None, **kwargs): def fit(self, X, y=None, sample_weight=None, client=None, **kwargs):
...@@ -503,7 +503,7 @@ class DaskLGBMRegressor(LGBMRegressor, _LGBMModel): ...@@ -503,7 +503,7 @@ class DaskLGBMRegressor(LGBMRegressor, _LGBMModel):
return self._to_local(LGBMRegressor) return self._to_local(LGBMRegressor)
class DaskLGBMRanker(LGBMRanker, _LGBMModel): class DaskLGBMRanker(LGBMRanker, _DaskLGBMModel):
"""Docstring is inherited from the lightgbm.LGBMRanker.""" """Docstring is inherited from the lightgbm.LGBMRanker."""
def fit(self, X, y=None, sample_weight=None, init_score=None, group=None, client=None, **kwargs): def fit(self, X, y=None, sample_weight=None, init_score=None, group=None, client=None, **kwargs):
......
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