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

Fix b_main output with multiple seeds

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2024-01-16 10:42:24 +01:00
parent 2b20d0315c
commit 5e1d59acdb
1 changed files with 9 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
src/Platform/modules/Experiment.cc
View File

@@ -159,9 +159,16 @@ namespace platform {
auto num_states = torch::zeros({ nResults }, torch::kFloat64); auto num_states = torch::zeros({ nResults }, torch::kFloat64);
Timer train_timer, test_timer; Timer train_timer, test_timer;
int item = 0; int item = 0;
bool first_seed = true;
for (auto seed : randomSeeds) { for (auto seed : randomSeeds) {
if (!quiet) if (!quiet) {
std::cout << "(" << seed << ") doing Fold: " << flush; string prefix = "";
if (!first_seed) {
prefix = "\n" + string(36, ' ');
}
std::cout << prefix << "(" << setw(4) << seed << ") doing Fold: " << flush;
first_seed = false;
}
folding::Fold* fold; folding::Fold* fold;
if (stratified) if (stratified)
fold = new folding::StratifiedKFold(nfolds, y, seed); fold = new folding::StratifiedKFold(nfolds, y, seed);