Begin Stratified KFold

src/Platform/Folding.h

@@ -7,12 +7,19 @@ private:
     int k;
     int n;
     vector<int> indices;
-
+    int seed;
 public:
-    KFold(int k, int n, int seed);
-    pair<vector<int>, vector<int>> getFold(int);
+    KFold(int k, int n, int seed = -1);
+    pair<vector<int>, vector<int>> getFold(int nFold);
 };
-class KStratifiedFold {
-
+class StratifiedKFold {
+private:
+    int k;
+    int n;
+    vector<vector<int>> stratified_indices;
+    unsigned seed;
+public:
+    StratifiedKFold(int k, const vector<int>& y, int seed = -1);
+    pair<vector<int>, vector<int>> getFold(int nFold);
 };
 #endif