From aed72e9c14bdbb4fa2f6024fbb2c79910f8deccd Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Ricardo=20Monta=C3=B1ana?= Date: Thu, 10 Mar 2022 19:55:51 +0100 Subject: [PATCH] Fix best report and grid report --- src/Results.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/src/Results.py b/src/Results.py index 2999ee5..203e563 100644 --- a/src/Results.py +++ b/src/Results.py @@ -183,11 +183,11 @@ class Report(BaseReport): class ReportBest(BaseReport): - header_lengths = [30, 8, 50, 35] + header_lengths = [30, 8, 76, 45] header_cols = [ "Dataset", "Score", - "File", + "File/Message", "Hyperparameters", ] @@ -197,6 +197,8 @@ class ReportBest(BaseReport): if best else Files.grid_output(score, model) ) + self.best = best + self.grid = grid file_name = os.path.join(Folders.results, name) super().__init__(file_name, best_file=True) self.compare = False @@ -225,8 +227,9 @@ class ReportBest(BaseReport): def header(self): self.header_line("*") + kind = "Best" if self.best else "Grid" self.header_line( - f" Report Best {self.score_name} Scores with {self.model} in any " + f" Report {kind} {self.score_name} Scores with {self.model} in any " "platform" ) self.header_line("*")