Add resport score for stree

update param_analysis for stree only
This commit is contained in:
2021-03-23 00:50:04 +01:00
parent 08fb237001
commit 47078208bc
4 changed files with 291 additions and 49 deletions

View File

@@ -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():