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

Add messages to control trace

Browse Source
This commit is contained in:
Ricardo Montañana Gómez 2023-12-14 21:06:43 +01:00
parent b73f4be146
commit ffe6d37436
Signed by: rmontanana
GPG Key ID: 46064262FD9A7ADE
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/Platform/GridSearch.cc
View File

@ -297,9 +297,11 @@ namespace platform {
total = new char[max_size * config_mpi.n_procs] {};
}
// 3.2 Gather all the results from the workers into the manager
std::cout << "(" << config_mpi.rank << ")" << msg << std::endl;
MPI_Gather(msg, max_size, MPI_CHAR, total, max_size * config_mpi.n_procs, MPI_CHAR, config_mpi.manager, MPI_COMM_WORLD);
delete[] msg;
if (config_mpi.rank == config_mpi.manager) {
std::cout << "Manager taking final control!" << std::endl;
json total_results;
json best_results;
// 3.3 Compile the results from all the workers
@ -312,6 +314,7 @@ namespace platform {
}
}
delete[] total;
std::cout << "Total results: " << total_results.dump() << std::endl;
// 3.4 Filter the best hyperparameters for each dataset
auto grid = GridData(Paths::grid_input(config.model));
for (auto& [dataset, folds] : total_results.items()) {
@ -332,8 +335,10 @@ namespace platform {
};
best_results[dataset] = result;
}
std::cout << "Best results: " << best_results.dump() << std::endl;
save(total_results);
}
std::cout << "Process " << config_mpi.rank << " finished!" << std::endl;
}
void GridSearch::go()
{