rmontanana/BayesNet
1
0
Fork 0
Code Issues 2 Pull Requests 1 Actions Packages Projects Releases 6 Wiki Activity

Complete Stratified K Fold

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2023-07-22 11:23:35 +02:00
parent f6e154bc6e
commit 41cceece20
5 changed files with 88 additions and 86 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/Platform/Folding.h
View File

@@ -6,8 +6,8 @@ class KFold {
private:
int k;
int n;
vector<int> indices;
int seed;
vector<int> indices;
public:
KFold(int k, int n, int seed = -1);
pair<vector<int>, vector<int>> getFold(int nFold);
@@ -16,8 +16,8 @@ class StratifiedKFold {
private:
int k;
int n;
int seed;
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);