22 lines
492 B
Python
22 lines
492 B
Python
"""Decision tree based models for classification and regression."""
|
|
|
|
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",
|
|
"plot_tree",
|
|
"export_text",
|
|
]
|