From cc14d338f81eb1ccbccd4f55e6e5c57726c0d611 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Ricardo=20Monta=C3=B1ana?= <rmontanana@gmail.com>
Date: Fri, 17 Dec 2021 13:21:35 +0100
Subject: [PATCH] Add version info to  Reports

---
 src/Experiments.py | 2 ++
 src/Results.py     | 6 ++++--
 2 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/src/Experiments.py b/src/Experiments.py
index 5fb86e8..81a29f3 100644
--- a/src/Experiments.py
+++ b/src/Experiments.py
@@ -208,6 +208,7 @@ class Experiment:
                 shuffle=True, random_state=random_state, n_splits=self.folds
             )
             clf = self._build_classifier(random_state, hyperparameters)
+            self.version = clf.version() if hasattr(clf, "version") else "-"
             with warnings.catch_warnings():
                 warnings.filterwarnings("ignore")
                 res = cross_validate(
@@ -248,6 +249,7 @@ class Experiment:
         output = {}
         output["score_name"] = self.score_name
         output["model"] = self.model_name
+        output["version"] = self.version
         output["stratified"] = self.stratified
         output["folds"] = self.folds
         output["date"] = self.date
diff --git a/src/Results.py b/src/Results.py
index 2db736d..02dd1dc 100644
--- a/src/Results.py
+++ b/src/Results.py
@@ -136,7 +136,8 @@ class Report(BaseReport):
             self._compare_totals = {}
         self.header_line("*")
         self.header_line(
-            f" Report {self.data['model']} with {self.data['folds']} Folds "
+            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"
         )
         self.header_line(
@@ -262,7 +263,8 @@ class Excel(BaseReport):
         self.sheet.write(
             0,
             0,
-            f" Report {self.data['model']} with {self.data['folds']} Folds "
+            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",
             header,
         )