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

Fix some lint warnings

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2023-07-29 20:37:51 +02:00
parent 5efa3beaee
commit 8b2ed26ab7
7 changed files with 34 additions and 51 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/Platform/Folding.h
View File

@@ -22,7 +22,7 @@ private:
vector<int> indices;
public:
KFold(int k, int n, int seed = -1);
pair<vector<int>, vector<int>> getFold(int nFold);
pair<vector<int>, vector<int>> getFold(int nFold) override;
};
class StratifiedKFold : public Fold {
private:
@@ -32,6 +32,6 @@ private:
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);
pair<vector<int>, vector<int>> getFold(int nFold) override;
};
#endif