scikit-learn/sklearn/tree/__init__.py

14 lines
441 B
Python
Raw Normal View History

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