Changed class exposure.

bsc-wdc · Aug 18, 2021 · 1ddfeed · 1ddfeed
1 parent 197ea85
commit 1ddfeed
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/dislib/trees/__init__.py b/dislib/trees/__init__.py
@@ -0,0 +1,14 @@
+from dislib.trees.forest import RandomForestClassifier, RandomForestRegressor
+from dislib.trees.decision_tree import (
+    DecisionTreeClassifier,
+    DecisionTreeRegressor,
+)
+from dislib.trees.data import transform_to_rf_dataset
+
+__all__ = [
+    "RandomForestClassifier",
+    "RandomForestRegressor",
+    "DecisionTreeClassifier",
+    "DecisionTreeRegressor",
+    "transform_to_rf_dataset",
+]