#3 Add degree hyperparam and update notebooks

Update readme to add new notebooks
2025-08-17 00:16:07 +00:00 · 2020-06-08 20:16:42 +02:00
parent d7c0bc3bc5
commit 26273e936a
8 changed files with 680 additions and 591 deletions
--- a/stree/Strees.py
+++ b/stree/Strees.py
@@ -126,6 +126,7 @@ class Stree(BaseEstimator, ClassifierMixin):
        random_state: int = None,
        max_depth: int = None,
        tol: float = 1e-4,
+        degree: int = 3,
        gamma="scale",
        min_samples_split: int = 0,
    ):
@@ -136,6 +137,7 @@ class Stree(BaseEstimator, ClassifierMixin):
        self.max_depth = max_depth
        self.tol = tol
        self.gamma = gamma
+        self.degree = degree
        self.min_samples_split = min_samples_split

    def _more_tags(self) -> dict:
@@ -263,6 +265,7 @@ class Stree(BaseEstimator, ClassifierMixin):
                tol=self.tol,
                C=self.C,
                gamma=self.gamma,
+                degree=self.degree,
            )
        )