Doctorado-ML/benchmark
1
0
Fork 0
mirror of https://github.com/Doctorado-ML/benchmark.git synced 2025-08-15 15:35:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix some errors in tests

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2024-12-16 00:53:11 +01:00
parent 2a2ed81a6c
commit 419c899c94
5 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
benchmark/Datasets.py
View File

@@ -152,7 +152,7 @@ class Datasets:
if dataset_name is None: if dataset_name is None:
return class_names, results return class_names, results
try: try:
class_name = class_names[sets.index(dataset_name)] class_name = class_names[results.index(dataset_name)]
except ValueError: except ValueError:
raise ValueError(f"Unknown dataset: {dataset_name}") raise ValueError(f"Unknown dataset: {dataset_name}")
return [class_name], [dataset_name] return [class_name], [dataset_name]

2
benchmark/tests/Models_test.py
View File

@@ -102,7 +102,7 @@ class ModelTest(TestBase):
test = { test = {
"STree": ((11, 6, 4), 1.0), "STree": ((11, 6, 4), 1.0),
"Wodt": ((303, 152, 50), 0.9382022471910112), "Wodt": ((303, 152, 50), 0.9382022471910112),
"ODTE": ((7.86, 4.43, 3.37), 1.0), "ODTE": ((786, 443, 337), 1.0),
"Cart": ((23, 12, 5), 1.0), "Cart": ((23, 12, 5), 1.0),
"SVC": ((0, 0, 0), 0.7078651685393258), "SVC": ((0, 0, 0), 0.7078651685393258),
"RandomForest": ((21.3, 11, 5.26), 1.0), "RandomForest": ((21.3, 11, 5.26), 1.0),

4
benchmark/tests/datasets/all.txt
View File

@@ -1,2 +1,2 @@
iris,class,all iris;class;all
wine,class,[0, 1] wine;class;[0, 1]

2
benchmark/tests/test_files/exreport_excel_Datasets.test
View File

@@ -1,4 +1,4 @@
1;1;"Datasets used in benchmark ver. 0.5.0" 1;1;"Datasets used in benchmark ver. 1.0.1"
2;1;" Default score accuracy" 2;1;" Default score accuracy"
2;2;"Cross validation" 2;2;"Cross validation"
2;6;"5 Folds" 2;6;"5 Folds"

2
benchmark/tests/test_files/report_datasets.test
View File

@@ -1,4 +1,4 @@
Datasets used in benchmark ver. 0.5.0 Datasets used in benchmark ver. 1.0.1
Dataset Sampl. Feat. Cont Cls Balance Dataset Sampl. Feat. Cont Cls Balance
============================== ====== ===== ==== === ========================================== ============================== ====== ===== ==== === ==========================================