25 lines
572 B
Python
25 lines
572 B
Python
|
"""Decision tree based models for classification and regression."""
|
||
|
|
|
||
|
|
# Authors: The scikit-learn developers
|
||
|
|
# SPDX-License-Identifier: BSD-3-Clause
|
||
|
|
|
||
|
|
from ._classes import (
|
||
|
|
BaseDecisionTree,
|
||
|
|
DecisionTreeClassifier,
|
||
|
|
DecisionTreeRegressor,
|
||
|
|
ExtraTreeClassifier,
|
||
|
|
ExtraTreeRegressor,
|
||
|
|
)
|
||
|
|
from ._export import export_graphviz, export_text, plot_tree
|
||
|
|
|
||
|
|
__all__ = [
|
||
|
|
"BaseDecisionTree",
|
||
|
|
"DecisionTreeClassifier",
|
||
|
|
"DecisionTreeRegressor",
|
||
|
|
"ExtraTreeClassifier",
|
||
|
|
"ExtraTreeRegressor",
|
||
|
|
"export_graphviz",
|
||
|
|
"export_text",
|
||
|
|
"plot_tree",
|
||
|
|
]
|