Skip to content

Commit 9af0865

Browse files
authored
Merge pull request #25 from shenkha/ensembletree-implementation
Small Fix on API Libraries
2 parents 213f612 + ddd5fdd commit 9af0865

File tree

2 files changed

+3
-0
lines changed

2 files changed

+3
-0
lines changed

docs/api/linear.rst

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,8 @@ The simplest usage is::
2626

2727
.. autofunction:: train_tree
2828

29+
.. autofunction:: train_ensemble_tree
30+
2931
.. autofunction:: predict_values
3032

3133
.. autofunction:: get_topk_labels

libmultilabel/linear/tree.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -427,6 +427,7 @@ def train_ensemble_tree(
427427
seed: int = None,
428428
) -> EnsembleTreeModel:
429429
"""Trains an ensemble of tree models (Parabel/Bonsai-style).
430+
430431
Args:
431432
y (sparse.csr_matrix): A 0/1 matrix with dimensions number of instances * number of classes.
432433
x (sparse.csr_matrix): A matrix with dimensions number of instances * number of features.

0 commit comments

Comments
 (0)