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

Update and fix reports

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2021-09-22 18:04:27 +02:00
parent 4ad61998f4
commit db771bf6f2
3 changed files with 11 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
report_score.py
View File

@@ -125,9 +125,10 @@ def process_dataset(dataset, verbose, model, params):
if model == "stree":
# Get the optimized parameters
record = dbh.find_best(dataset, model, "gridsearch")
hyperparameters = json.loads(record[8] if record[8] != "" else "{}")
hyperparameters = json.loads(
record[8].replace('\\"', '"') if record[8] != "" else "{}"
)
hyperparameters.pop("random_state", None)
print("*" * 100)
for random_state in random_seeds:
random.seed(random_state)
np.random.seed(random_state)