Merge branch 'main' of github.com:doctorado-ml/benchmark into main

This commit is contained in:
2022-03-17 23:11:06 +01:00
3 changed files with 28 additions and 7 deletions

File diff suppressed because one or more lines are too long

File diff suppressed because one or more lines are too long

View File

@@ -275,6 +275,7 @@ class Excel(BaseReport):
if book is None: if book is None:
self.excel_file_name = self.file_name.replace(".json", ".xlsx") self.excel_file_name = self.file_name.replace(".json", ".xlsx")
self.book = xlsxwriter.Workbook(self.excel_file_name) self.book = xlsxwriter.Workbook(self.excel_file_name)
self.set_book_properties()
self.close = True self.close = True
else: else:
self.book = book self.book = book
@@ -283,6 +284,30 @@ class Excel(BaseReport):
self.max_hyper_width = 0 self.max_hyper_width = 0
self.col_hyperparams = 0 self.col_hyperparams = 0
@staticmethod
def set_properties(book, title):
book.set_properties(
{
"title": title,
"subject": "Machine learning results",
"author": "Ricardo Montañana Gómez",
"manager": "Dr. J. A. Gámez, Dr. J. M. Puerta",
"company": "UCLM",
"comments": "Created with Python and XlsxWriter",
}
)
def set_book_properties(self):
self.set_properties(self.book, self.get_title())
def get_title(self):
return (
f" Report {self.data['model']} ver. {self.data['version']}"
f" with {self.data['folds']} Folds "
f"cross validation and {len(self.data['seeds'])} random seeds. "
f"{self.data['date']} {self.data['time']}"
)
def get_file_name(self): def get_file_name(self):
return self.excel_file_name return self.excel_file_name
@@ -317,12 +342,7 @@ class Excel(BaseReport):
"bg_color": self.color1, "bg_color": self.color1,
} }
) )
header_text = ( header_text = self.get_title()
f" Report {self.data['model']} ver. {self.data['version']}"
f" with {self.data['folds']} Folds "
f"cross validation and {len(self.data['seeds'])} random seeds. "
f"{self.data['date']} {self.data['time']}"
)
self.sheet.merge_range(0, 0, 0, 11, header_text, merge_format) self.sheet.merge_range(0, 0, 0, 11, header_text, merge_format)
self.sheet.merge_range( self.sheet.merge_range(
1, 0, 1, 11, f" {self.data['title']}", merge_format_subheader 1, 0, 1, 11, f" {self.data['title']}", merge_format_subheader
@@ -699,6 +719,7 @@ class Benchmark:
def excel(self): def excel(self):
book = xlsxwriter.Workbook(self.get_excel_file_name()) book = xlsxwriter.Workbook(self.get_excel_file_name())
Excel.set_properties(book, "Experimentation summary")
sheet = book.add_worksheet("Benchmark") sheet = book.add_worksheet("Benchmark")
normal = book.add_format({"font_size": 14, "border": 1}) normal = book.add_format({"font_size": 14, "border": 1})
decimal = book.add_format( decimal = book.add_format(