rmontanana/Folding
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 4 Wiki Activity

Use another random device

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2024-04-03 00:22:30 +02:00
parent 78f2f5b299
commit 96327e7ee0
71 changed files with 25201 additions and 25201 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/TestFolding.cc
View File

@@ -27,7 +27,7 @@ TEST_CASE("KFold Test", "[Platform][KFold]")
auto fname = "kfold_" + file_name + "_" + std::to_string(i) + ".csv";
auto indices = train_indices;
indices.insert(indices.end(), test_indices.begin(), test_indices.end());
//CSVFiles::write_csv(fname, indices);
CSVFiles::write_csv(fname, indices);
auto expected_indices = CSVFiles::read_csv(fname);
REQUIRE(indices == expected_indices);
bool result = train_indices.size() == number || train_indices.size() == number + 1;
@@ -77,7 +77,7 @@ TEST_CASE("StratifiedKFold Test", "[Platform][StratifiedKFold]")
auto fname = "stratkfold_" + file_name + "_" + std::to_string(nFolds) + "_" + std::to_string(fold) + ".csv";
auto indices = train_indicesv;
indices.insert(indices.end(), test_indicesv.begin(), test_indicesv.end());
//CSVFiles::write_csv(fname, indices);
CSVFiles::write_csv(fname, indices);
auto expected_indices = CSVFiles::read_csv(fname);
REQUIRE(indices == expected_indices);
// In the worst case scenario, the number of samples in the training set is number + raw.classNumStates