15 lines
476 B
Python
15 lines
476 B
Python
"""
|
|
The :mod:`sklearn.tree` module includes decision tree-based models for
|
|
classification and regression.
|
|
"""
|
|
|
|
from .tree import DecisionTreeClassifier
|
|
from .tree import DecisionTreeRegressor
|
|
from .tree import ExtraTreeClassifier
|
|
from .tree import ExtraTreeRegressor
|
|
from .export import export_graphviz, plot_tree
|
|
|
|
__all__ = ["DecisionTreeClassifier", "DecisionTreeRegressor",
|
|
"ExtraTreeClassifier", "ExtraTreeRegressor", "export_graphviz",
|
|
"plot_tree"]
|