update inference.tree._tree with sklearn 1.15.2

uber · Dec 6, 2024 · 5390458 · 5390458
1 parent a52eef4
commit 5390458
Show file tree

Hide file tree

Showing 19 changed files with 9,400 additions and 2,745 deletions.
diff --git a/causalml/inference/tree/_tree/__init__.py b/causalml/inference/tree/_tree/__init__.py
@@ -1,4 +1,26 @@
 """
 This part of tree structures definition was initially borrowed from
-https://github.com/scikit-learn/scikit-learn/tree/1.0.2/sklearn/tree
+https://github.com/scikit-learn/scikit-learn/tree/1.5.2/sklearn/tree
 """
+
+"""Decision tree based models for classification and regression."""
+
+from ._classes import (
+    BaseDecisionTree,
+    DecisionTreeClassifier,
+    DecisionTreeRegressor,
+    ExtraTreeClassifier,
+    ExtraTreeRegressor,
+)
+from ._export import export_graphviz, export_text, plot_tree
+
+__all__ = [
+    "BaseDecisionTree",
+    "DecisionTreeClassifier",
+    "DecisionTreeRegressor",
+    "ExtraTreeClassifier",
+    "ExtraTreeRegressor",
+    "export_graphviz",
+    "plot_tree",
+    "export_text",
+]