You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
25 lines
596 B
25 lines
596 B
"""Decision tree based models for classification and regression."""
|
|
|
|
# Authors: The scikit-learn developers
|
|
# SPDX-License-Identifier: BSD-3-Clause
|
|
|
|
from sklearn.tree._classes import (
|
|
BaseDecisionTree,
|
|
DecisionTreeClassifier,
|
|
DecisionTreeRegressor,
|
|
ExtraTreeClassifier,
|
|
ExtraTreeRegressor,
|
|
)
|
|
from sklearn.tree._export import export_graphviz, export_text, plot_tree
|
|
|
|
__all__ = [
|
|
"BaseDecisionTree",
|
|
"DecisionTreeClassifier",
|
|
"DecisionTreeRegressor",
|
|
"ExtraTreeClassifier",
|
|
"ExtraTreeRegressor",
|
|
"export_graphviz",
|
|
"export_text",
|
|
"plot_tree",
|
|
]
|