14 lines
439 B
Python
14 lines
439 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 .tree import export_graphviz
|
|
|
|
__all__ = ["DecisionTreeClassifier", "DecisionTreeRegressor",
|
|
"ExtraTreeClassifier", "ExtraTreeRegressor", "export_graphviz"]
|