From 2a2ed81a6c1fa8b5dfd653b2dba30dcdaf1a08bc Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Ricardo=20Monta=C3=B1ana=20G=C3=B3mez?=
 <rmontanana@gmail.com>
Date: Sat, 14 Dec 2024 23:50:58 +0100
Subject: [PATCH] Fix Arff datasets mistake Fix table_report partial mistake

---
 benchmark/Datasets.py                         | 31 ++++++++++---------
 .../partials/table_report_bootstrap.html      |  2 +-
 .../partials/table_report_bulma.html          |  2 +-
 3 files changed, 18 insertions(+), 17 deletions(-)

diff --git a/benchmark/Datasets.py b/benchmark/Datasets.py
index ebb5e52..4909f0c 100644
--- a/benchmark/Datasets.py
+++ b/benchmark/Datasets.py
@@ -32,6 +32,8 @@ class DatasetsArff:
     def get_range_features(X, c_features):
         if c_features.strip() == "all":
             return list(range(X.shape[1]))
+        if c_features.strip() == "none":
+            return []
         return json.loads(c_features)
 
     def load(self, name, class_name):
@@ -129,27 +131,26 @@ class Datasets:
 
     def _init_names(self, dataset_name):
         file_name = os.path.join(self.dataset.folder(), Files.index)
-        default_class = "class"
         self.continuous_features = {}
         with open(file_name) as f:
             sets = f.read().splitlines()
             sets = [x for x in sets if not x.startswith("#")]
-            class_names = [default_class] * len(sets)
-        if "," in sets[0]:
-            result = []
-            class_names = []
-            for data in sets:
-                name, class_name, features = data.split(",", 2)
-                result.append(name)
-                class_names.append(class_name)
-                self.continuous_features[name] = features
-            sets = result
-        else:
-            for name in sets:
-                self.continuous_features[name] = None
+        results = []
+        class_names = []
+        for set_name in sets:
+            try:
+                name, class_name, features = set_name.split(";")
+            except ValueError:
+                class_name = "class"
+                features = "all"
+                name = set_name
+            results.append(name)
+            class_names.append(class_name)
+            features = features.strip()
+            self.continuous_features[name] = features
         # Set as dataset list the dataset passed as argument
         if dataset_name is None:
-            return class_names, sets
+            return class_names, results
         try:
             class_name = class_names[sets.index(dataset_name)]
         except ValueError:
diff --git a/benchmark/scripts/app/templates/partials/table_report_bootstrap.html b/benchmark/scripts/app/templates/partials/table_report_bootstrap.html
index 8be1437..8228692 100644
--- a/benchmark/scripts/app/templates/partials/table_report_bootstrap.html
+++ b/benchmark/scripts/app/templates/partials/table_report_bootstrap.html
@@ -88,7 +88,7 @@
         <button type="button"
                 class="btn-close"
                 aria-label="Close"
-                onclick="location.href = '/index/{{ compare }}'"></button>
+                onclick="location.href = '{{ back }}'"></button>
         <h7>
         <b>
           Total score: {{ "%.6f" % (data.results | sum(attribute="score") ) }}
diff --git a/benchmark/scripts/app/templates/partials/table_report_bulma.html b/benchmark/scripts/app/templates/partials/table_report_bulma.html
index 07e87db..a6b53d3 100644
--- a/benchmark/scripts/app/templates/partials/table_report_bulma.html
+++ b/benchmark/scripts/app/templates/partials/table_report_bulma.html
@@ -90,7 +90,7 @@
         {% endif %}
         <h2 class="has-text-white has-background-primary">
           <b>
-            <button class="delete" onclick="location.href = '/index/{{ compare }}'"></button>
+            <button class="delete" onclick="location.href = '{{ back }}'"></button>
             Total score: {{ "%.6f" % (data.results | sum(attribute="score") ) }}
           </b>
         </h2>