Ensemble Experiment, Folding, Classifiers and Network together

src/Platform/Folding.h

@@ -1,5 +1,6 @@
 #ifndef FOLDING_H
 #define FOLDING_H
+#include <torch/torch.h>
 #include <vector>
 using namespace std;
 
@@ -12,6 +13,7 @@ public:
     Fold(int k, int n, int seed = -1) : k(k), n(n), seed(seed) {}
     virtual pair<vector<int>, vector<int>> getFold(int nFold) = 0;
     virtual ~Fold() = default;
+    int getNumberOfFolds() { return k; }
 };
 class KFold : public Fold {
 private:
@@ -21,9 +23,13 @@ public:
     pair<vector<int>, vector<int>> getFold(int nFold);
 };
 class StratifiedKFold : public Fold {
+private:
+    vector<int> y;
     vector<vector<int>> stratified_indices;
+    void build();
 public:
     StratifiedKFold(int k, const vector<int>& y, int seed = -1);
+    StratifiedKFold(int k, torch::Tensor& y, int seed = -1);
     pair<vector<int>, vector<int>> getFold(int nFold);
 };
 #endif