25 lines
534 B
Python
25 lines
534 B
Python
"""
|
|
The :mod:`sklearn.tree` module includes 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",
|
|
]
|