Implements fit and predict methods for bagging (mars-project#2516)

Marascax · Oct 19, 2021 · 3c67be0 · 3c67be0
1 parent a5e4044
commit 3c67be0
Show file tree

Hide file tree

Showing 6 changed files with 1,063 additions and 12 deletions.
diff --git a/docs/source/reference/learn/reference.rst b/docs/source/reference/learn/reference.rst
@@ -91,6 +91,8 @@ Ensemble Methods
 .. autosummary::
    :toctree: generated/
 
+   ensemble.BaggingClassifier
+   ensemble.BaggingRegressor
    ensemble.BlockwiseVotingClassifier
    ensemble.BlockwiseVotingRegressor
 

diff --git a/mars/core/operand/core.py b/mars/core/operand/core.py
@@ -196,7 +196,7 @@ def _create_tileable(self,
 
     def _new_tileables(self,
                        inputs: List[TileableType],
-                       kws: dict = None,
+                       kws: List[dict] = None,
                        **kw) -> List[TileableType]:
         output_limit = kw.pop('output_limit', None)
         if output_limit is None:
@@ -227,7 +227,7 @@ def _new_tileables(self,
 
     def new_tileables(self,
                       inputs: List[TileableType],
-                      kws=None,
+                      kws: List[dict] = None,
                       **kw) -> List[TileableType]:
         """
         Create tileable objects(Tensors or DataFrames).
@@ -240,7 +240,7 @@ def new_tileables(self,
         ----------
         inputs : list
             Input tileables
-        kws : dict
+        kws : List[dict]
             Kwargs for each output.
         kw : dict
             Common kwargs for all outputs.

diff --git a/mars/learn/ensemble/__init__.py b/mars/learn/ensemble/__init__.py
@@ -12,5 +12,5 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from ._bagging import BaggingSample, BaggingSampleReindex
+from ._bagging import BaggingClassifier, BaggingRegressor
 from ._blockwise import BlockwiseVotingClassifier, BlockwiseVotingRegressor