mirror of
https://github.com/Doctorado-ML/STree.git
synced 2025-08-16 16:06:01 +00:00
New_version_sklearn (#56)
* test: 🧪 Update max_iter as int in test_multiclass_dataset * refactor: 📝 Rename base_estimator to estimator as the former is deprectated in notebook * refactor: 📌 Convert max_iter to int as needed in sklearn 1.2 * chore: 🔖 Update version info to 1.3.1
This commit is contained in:
committed by
GitHub
parent
c37f044e3a
commit
5b791bc5bf
@@ -139,7 +139,7 @@ class Stree(BaseEstimator, ClassifierMixin):
|
||||
self,
|
||||
C: float = 1.0,
|
||||
kernel: str = "linear",
|
||||
max_iter: int = 1e5,
|
||||
max_iter: int = int(1e5),
|
||||
random_state: int = None,
|
||||
max_depth: int = None,
|
||||
tol: float = 1e-4,
|
||||
|
@@ -1 +1 @@
|
||||
__version__ = "1.3.0"
|
||||
__version__ = "1.3.1"
|
||||
|
@@ -306,7 +306,7 @@ class Stree_test(unittest.TestCase):
|
||||
for criteria in ["max_samples", "impurity"]:
|
||||
for kernel in self._kernels:
|
||||
clf = Stree(
|
||||
max_iter=1e4,
|
||||
max_iter=int(1e4),
|
||||
multiclass_strategy="ovr"
|
||||
if kernel == "liblinear"
|
||||
else "ovo",
|
||||
|
Reference in New Issue
Block a user