Doctorado-ML/odte
1
1
Fork 0
mirror of https://github.com/Doctorado-ML/Odte.git synced 2025-07-11 08:12:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Set default estimator to STree

Browse Source
This commit is contained in:
Ricardo Montañana Gómez 2024-07-28 18:30:17 +02:00
parent 02e75b3c3e
commit b19264b1eb
Signed by: rmontanana
GPG Key ID: 46064262FD9A7ADE
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
odte/Odte.py
View File

@ -5,6 +5,7 @@ __license__ = "MIT"
Build a forest of oblique trees based on STree, admits any base classifier Build a forest of oblique trees based on STree, admits any base classifier
as well as well
""" """
from __future__ import annotations from __future__ import annotations
import random import random
import json import json
@ -31,7 +32,7 @@ class Odte(BaseEnsemble, ClassifierMixin):
self, self,
# n_jobs = -1 to use all available cores # n_jobs = -1 to use all available cores
n_jobs: int = -1, n_jobs: int = -1,
estimator: BaseEstimator = None, estimator: BaseEstimator = Stree(),
random_state: int = 0, random_state: int = 0,
max_features: Optional[Union[str, int, float]] = None, max_features: Optional[Union[str, int, float]] = None,
max_samples: Optional[Union[int, float]] = None, max_samples: Optional[Union[int, float]] = None,