Set default estimator to STree

Doctorado-ML · Jul 28, 2024 · b19264b · b19264b
1 parent 02e75b3
commit b19264b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/odte/Odte.py b/odte/Odte.py
@@ -5,6 +5,7 @@
 Build a forest of oblique trees based on STree, admits any base classifier
 as well
 """
+
 from __future__ import annotations
 import random
 import json
@@ -31,7 +32,7 @@ def __init__(
         self,
         # n_jobs = -1 to use all available cores
         n_jobs: int = -1,
-        estimator: BaseEstimator = None,
+        estimator: BaseEstimator = Stree(),
         random_state: int = 0,
         max_features: Optional[Union[str, int, float]] = None,
         max_samples: Optional[Union[int, float]] = None,