Fix b_main output with multiple seeds

This commit is contained in:
2024-01-16 10:42:24 +01:00
parent 2b20d0315c
commit 5e1d59acdb

View File

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