scikit-learn/sklearn/tree/__init__.py

11 lines
217 B
Python

"""
Tree-based models
=================
sklearn.tree is a module to fit data to decision trees.
"""
from .tree import DecisionTreeClassifier
from .tree import DecisionTreeRegressor
from .tree import export_graphviz