rename the tree.py and export.py and make the tests pass

adrinjalali · adrinjalali · commit df7f54d4d5ec · 2019-09-11T18:32:19.000+02:00
diff --git a/sklearn/ensemble/gradient_boosting.py b/sklearn/ensemble/gradient_boosting.py
@@ -44,7 +44,7 @@
 
 from time import time
 from ..model_selection import train_test_split
-from ..tree.tree import DecisionTreeRegressor
+from ..tree import DecisionTreeRegressor
 from ..tree._tree import DTYPE, DOUBLE
 from ..tree._tree import TREE_LEAF
 from . import _gb_losses
diff --git a/sklearn/tree/__init__.py b/sklearn/tree/__init__.py
@@ -3,11 +3,11 @@
 classification and regression.
 """
 
-from .tree import DecisionTreeClassifier
-from .tree import DecisionTreeRegressor
-from .tree import ExtraTreeClassifier
-from .tree import ExtraTreeRegressor
-from .export import export_graphviz, plot_tree, export_text
+from ._tree_base import DecisionTreeClassifier
+from ._tree_base import DecisionTreeRegressor
+from ._tree_base import ExtraTreeClassifier
+from ._tree_base import ExtraTreeRegressor
+from ._export import export_graphviz, plot_tree, export_text
 
 __all__ = ["DecisionTreeClassifier", "DecisionTreeRegressor",
            "ExtraTreeClassifier", "ExtraTreeRegressor", "export_graphviz",
diff --git a/sklearn/tree/_export.py b/sklearn/tree/_export.py
@@ -832,7 +832,7 @@ def export_text(decision_tree, feature_names=None, max_depth=10,
 
     >>> from sklearn.datasets import load_iris
     >>> from sklearn.tree import DecisionTreeClassifier
-    >>> from sklearn.tree.export import export_text
+    >>> from sklearn.tree import export_text
     >>> iris = load_iris()
     >>> X = iris['data']
     >>> y = iris['target']
diff --git a/sklearn/tree/_tree_base.py b/sklearn/tree/_tree_base.py
diff --git a/sklearn/tree/tests/test_tree.py b/sklearn/tree/tests/test_tree.py
@@ -38,8 +38,8 @@
 
 from sklearn import tree
 from sklearn.tree._tree import TREE_LEAF, TREE_UNDEFINED
-from sklearn.tree.tree import CRITERIA_CLF
-from sklearn.tree.tree import CRITERIA_REG
+from sklearn.tree._tree_base import CRITERIA_CLF
+from sklearn.tree._tree_base import CRITERIA_REG
 from sklearn import datasets
 
 from sklearn.utils import compute_sample_weight
