diff --git a/docs/api/linear.rst b/docs/api/linear.rst index 23361a6b..888db896 100644 --- a/docs/api/linear.rst +++ b/docs/api/linear.rst @@ -26,6 +26,8 @@ The simplest usage is:: .. autofunction:: train_tree +.. autofunction:: train_ensemble_tree + .. autofunction:: predict_values .. autofunction:: get_topk_labels diff --git a/libmultilabel/linear/tree.py b/libmultilabel/linear/tree.py index 34f06370..2e2d83df 100644 --- a/libmultilabel/linear/tree.py +++ b/libmultilabel/linear/tree.py @@ -426,6 +426,7 @@ def train_ensemble_tree( seed: int = None, ) -> EnsembleTreeModel: """Trains an ensemble of tree models (Parabel/Bonsai-style). + Args: y (sparse.csr_matrix): A 0/1 matrix with dimensions number of instances * number of classes. x (sparse.csr_matrix): A matrix with dimensions number of instances * number of features.