2011-08-02 18:28:05 +08:00
|
|
|
"""
|
2011-11-30 18:05:10 +08:00
|
|
|
The :mod:`sklearn.tree` module includes decision tree-based models for
|
2011-11-12 19:19:16 +08:00
|
|
|
classification and regression.
|
2011-08-02 18:28:05 +08:00
|
|
|
"""
|
2011-11-07 17:54:14 +08:00
|
|
|
|
2011-08-02 18:28:05 +08:00
|
|
|
from .tree import DecisionTreeClassifier
|
2011-08-11 18:25:57 +08:00
|
|
|
from .tree import DecisionTreeRegressor
|
2011-11-12 18:29:38 +08:00
|
|
|
from .tree import ExtraTreeClassifier
|
|
|
|
|
from .tree import ExtraTreeRegressor
|
2013-04-14 21:14:44 +08:00
|
|
|
from .export import export_graphviz
|
2012-12-22 20:02:50 +08:00
|
|
|
|
|
|
|
|
__all__ = ["DecisionTreeClassifier", "DecisionTreeRegressor",
|
|
|
|
|
"ExtraTreeClassifier", "ExtraTreeRegressor", "export_graphviz"]
|