Add header to grid output and report

This commit is contained in:
2023-12-01 10:30:53 +01:00
parent c460ef46ed
commit 33cd32c639
3 changed files with 117 additions and 23 deletions

View File

@@ -17,19 +17,22 @@ namespace platform {
bool only; // used with continue_from to only compute that dataset
bool discretize;
bool stratified;
int nested;
int n_folds;
std::vector<int> seeds;
};
class GridSearch {
public:
explicit GridSearch(struct ConfigGrid& config);
void go();
void goSingle();
void goNested();
~GridSearch() = default;
json getResults();
private:
void save(json& results) const;
json initializeResults();
vector<std::string> processDatasets(Datasets& datasets);
double processFile(std::string fileName, Datasets& datasets, HyperParameters& hyperparameters);
double processFileSingle(std::string fileName, Datasets& datasets, HyperParameters& hyperparameters);
struct ConfigGrid config;
};
} /* namespace platform */