Fix stratified folding mistake in remainders

2023-07-27 16:51:27 +02:00
parent 3d8fea7a37
commit 7f7ddad36a
8 changed files with 18 additions and 8 deletions
--- a/src/Platform/Datasets.h
+++ b/src/Platform/Datasets.h
@@ -58,6 +58,7 @@ namespace platform {
        pair<vector<vector<float>>&, vector<int>&> getVectors(string name);
        pair<vector<vector<int>>&, vector<int>&> getVectorsDiscretized(string name);
        pair<torch::Tensor&, torch::Tensor&> getTensors(string name);
+        bool isDataset(string name);
    };
 };