From b19264b1eb82a17876cb5cfc88b9190d4587fa83 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Ricardo=20Monta=C3=B1ana=20G=C3=B3mez?=
 <rmontanana@gmail.com>
Date: Sun, 28 Jul 2024 18:30:17 +0200
Subject: [PATCH] Set default estimator to STree

---
 odte/Odte.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/odte/Odte.py b/odte/Odte.py
index 179cf5b..4022f5f 100644
--- a/odte/Odte.py
+++ b/odte/Odte.py
@@ -5,6 +5,7 @@ __license__ = "MIT"
 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 @@ class Odte(BaseEnsemble, ClassifierMixin):
         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,