Doctorado-ML/stree
1
1
Fork 0
mirror of https://github.com/Doctorado-ML/STree.git synced 2025-08-17 08:26:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

Reformat test with new black version

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2024-03-05 18:46:19 +01:00
parent fb8b9b344f
commit 5f8ca8f3bb
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
stree/tests/Stree_test.py
View File

@@ -308,9 +308,9 @@ class Stree_test(unittest.TestCase):
for kernel in self._kernels: for kernel in self._kernels:
clf = Stree( clf = Stree(
max_iter=int(1e4), max_iter=int(1e4),
multiclass_strategy="ovr" multiclass_strategy=(
if kernel == "liblinear" "ovr" if kernel == "liblinear" else "ovo"
else "ovo", ),
kernel=kernel, kernel=kernel,
random_state=self._random_state, random_state=self._random_state,
) )