We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
2 parents 213f612 + ddd5fdd commit 9af0865Copy full SHA for 9af0865
docs/api/linear.rst
@@ -26,6 +26,8 @@ The simplest usage is::
26
27
.. autofunction:: train_tree
28
29
+.. autofunction:: train_ensemble_tree
30
+
31
.. autofunction:: predict_values
32
33
.. autofunction:: get_topk_labels
libmultilabel/linear/tree.py
@@ -427,6 +427,7 @@ def train_ensemble_tree(
427
seed: int = None,
428
) -> EnsembleTreeModel:
429
"""Trains an ensemble of tree models (Parabel/Bonsai-style).
430
431
Args:
432
y (sparse.csr_matrix): A 0/1 matrix with dimensions number of instances * number of classes.
433
x (sparse.csr_matrix): A matrix with dimensions number of instances * number of features.
0 commit comments