mirror of
https://github.com/Doctorado-ML/Odte.git
synced 2025-07-11 16:22:00 +00:00
Set default estimator to STree
This commit is contained in:
parent
02e75b3c3e
commit
b19264b1eb
@ -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,
|
||||||
|
Loading…
x
Reference in New Issue
Block a user