Continue with grid_experiment refactor

This commit is contained in:
2024-12-21 14:18:47 +01:00
parent 1cc19a7b19
commit 0318dcf8e5
7 changed files with 49 additions and 119 deletions

View File

@@ -4,7 +4,6 @@
#include <folding.hpp>
#include "main/Models.h"
#include "common/Paths.h"
#include "common/Colors.h"
#include "common/Utils.h"
#include "GridSearch.h"
@@ -55,6 +54,16 @@ namespace platform {
}
json GridSearch::build_tasks()
{
/*
* Each task is a json object with the following structure:
* {
* "dataset": "dataset_name",
* "idx_dataset": idx_dataset, // used to identify the dataset in the results
* // this index is relative to the list of used datasets in the actual run not to the whole datasets list
* "seed": # of seed to use,
* "fold": # of fold to process
* }
*/
auto tasks = json::array();
auto grid = GridData(Paths::grid_input(config.model));
auto datasets = Datasets(false, Paths::datasets());