diff --git a/Postprocessing/Report.py b/Postprocessing/Report.py index cc2578f4655f45d6179132627acd7c3cecf43f91..c0eac6f8e21f1be4dd28b3dabb29446328852e71 100644 --- a/Postprocessing/Report.py +++ b/Postprocessing/Report.py @@ -28,7 +28,7 @@ def generate_report_figures(map, palette_fn, results, summary, out_dir, map_name of['conf_matrices'] = [] for i,r in enumerate(results): cm = ConfusionMatrixDisplay.from_predictions(r['true_vs_pred'][0], r['true_vs_pred'][1], - normalize='true', include_values=True) + normalize='true', include_values=True, values_format='.2f') of['conf_matrices'].append('{}/conf_matrix_{}.png'.format(out_dir, str(i).zfill(2))) cm.ax_.set_xticklabels(class_names, rotation=45, ha='right') cm.ax_.set_yticklabels(class_names) @@ -157,7 +157,7 @@ def generate_pdf(of, out_pdf, name='output'): for k in list(rep.keys())[:-3] + list(rep.keys())[-2:]: table_data.append([k, rep[k]['precision'], rep[k]['recall'], rep[k]['f1-score'], rep[k]['support']]) pdf.set_xy(0, 150) - pdf.set_font("helvetica", '', 10) + pdf.set_font("helvetica", '', 6) with pdf.table(width=140, col_widths=(60,20,20,20,20)) as table: row = table.row() for datum in table_data[0]: