From 2d99b97432d9868fcc28c839ce8431ac8d77b0ab Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Ricardo=20Montan=CC=83ana?= <rmontanana@gmail.com>
Date: Sun, 16 Jan 2022 18:55:10 +0100
Subject: [PATCH] Add color to datasets report

---
 src/report.py | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/src/report.py b/src/report.py
index fb1090b..bd6ca0c 100644
--- a/src/report.py
+++ b/src/report.py
@@ -2,7 +2,7 @@ import argparse
 import numpy as np
 from Experiments import Datasets
 from Results import Report, Excel, SQL, ReportBest
-from Utils import Files
+from Utils import Files, TextColor
 
 """Build report on screen of a result file, optionally generate excel and sql
 file, and can compare results of report with best results obtained by model
@@ -69,10 +69,18 @@ def parse_arguments():
 
 def default_report():
     sets = Datasets()
+    color_line = TextColor.LINE1
+    print(color_line, end="")
     print(f"{'Dataset':30s} Samp. Feat Cls")
     print("=" * 30 + " ===== ==== ===")
     for line in sets:
         X, y = sets.load(line)
+        color_line = (
+            TextColor.LINE2
+            if color_line == TextColor.LINE1
+            else TextColor.LINE1
+        )
+        print(color_line, end="")
         print(
             f"{line:30s} {X.shape[0]:5,d} {X.shape[1]:4d} "
             f"{len(np.unique(y)):3d}"