Add report of grid search results and fix best results report

This commit is contained in:
2022-03-10 18:58:11 +01:00
parent 0d405a8556
commit 31a70fbd69
3 changed files with 501 additions and 13 deletions

View File

@@ -191,10 +191,13 @@ class ReportBest(BaseReport):
"Hyperparameters",
]
def __init__(self, score, model):
file_name = os.path.join(
Folders.results, Files.best_results(score, model)
def __init__(self, score, model, best, grid):
name = (
Files.best_results(score, model)
if best
else Files.grid_output(score, model)
)
file_name = os.path.join(Folders.results, name)
super().__init__(file_name, best_file=True)
self.compare = False
self.score_name = score