mirror of
https://github.com/Doctorado-ML/Stree_datasets.git
synced 2025-08-15 23:46:03 +00:00
Add resport score for stree
update param_analysis for stree only
This commit is contained in:
@@ -14,6 +14,7 @@ class Aggregation:
|
||||
self._dbh = dbh
|
||||
self._report = {}
|
||||
self._model_names = ["stree", "adaBoost", "bagging", "odte"]
|
||||
self._model_names = ["stree"]
|
||||
self._kernel_names = kernel_names
|
||||
|
||||
def find_values(self, dataset, parameter):
|
||||
@@ -35,10 +36,13 @@ class Aggregation:
|
||||
print("Aggregating data of best results ...")
|
||||
for dataset in dt:
|
||||
if result := self._dbh.find_best(dataset[0]):
|
||||
json_string = result[8] if result[8] != "" else "{}"
|
||||
accuracy = result[5]
|
||||
expected = result[10]
|
||||
model = result[3]
|
||||
json_result = json.loads(result[8])
|
||||
if model != "stree":
|
||||
continue
|
||||
json_result = json.loads(json_string)
|
||||
if "kernel" in json_result.keys():
|
||||
kernel = json_result["kernel"]
|
||||
elif "base_estimator__kernel" in json_result.keys():
|
||||
|
Reference in New Issue
Block a user