Add labels to confusion_matrices in results

lib/Files/ArffFiles.cc

@@ -155,12 +155,14 @@ std::string ArffFiles::trim(const std::string& source)
 std::vector<int> ArffFiles::factorize(const std::vector<std::string>& labels_t)
 {
     std::vector<int> yy;
+    labels.clear();
     yy.reserve(labels_t.size());
     std::map<std::string, int> labelMap;
     int i = 0;
     for (const std::string& label : labels_t) {
         if (labelMap.find(label) == labelMap.end()) {
             labelMap[label] = i++;
+            labels.push_back(label);
         }
         yy.push_back(labelMap[label]);
     }

lib/Files/ArffFiles.h

@@ -5,15 +5,6 @@
 #include <vector>
 
 class ArffFiles {
-private:
-    std::vector<std::string> lines;
-    std::vector<std::pair<std::string, std::string>> attributes;
-    std::string className;
-    std::string classType;
-    std::vector<std::vector<float>> X;
-    std::vector<int> y;
-    void generateDataset(int);
-    void loadCommon(std::string);
 public:
     ArffFiles();
     void load(const std::string&, bool = true);
@@ -22,11 +13,22 @@ public:
     unsigned long int getSize() const;
     std::string getClassName() const;
     std::string getClassType() const;
+    std::vector<std::string> getLabels() const { return labels; }
     static std::string trim(const std::string&);
     std::vector<std::vector<float>>& getX();
     std::vector<int>& getY();
     std::vector<std::pair<std::string, std::string>> getAttributes() const;
-    static std::vector<int> factorize(const std::vector<std::string>& labels_t);
+    std::vector<int> factorize(const std::vector<std::string>& labels_t);
+private:
+    std::vector<std::string> lines;
+    std::vector<std::pair<std::string, std::string>> attributes;
+    std::string className;
+    std::string classType;
+    std::vector<std::vector<float>> X;
+    std::vector<int> y;
+    std::vector<std::string> labels;
+    void generateDataset(int);
+    void loadCommon(std::string);
 };
 
 #endif