Build gridsearch structure

This commit is contained in:
2023-11-20 23:32:34 +01:00
parent 5876be4b24
commit 4628e48d3c
13 changed files with 197 additions and 33 deletions

30
src/Platform/GridSearch.h Normal file
View File

@@ -0,0 +1,30 @@
#ifndef GRIDSEARCH_H
#define GRIDSEARCH_H
#include <string>
#include <vector>
namespace platform {
struct ConfigGrid {
std::string model;
std::string score;
std::string path;
std::string input_file;
std::string output_file;
bool discretize;
bool stratified;
int n_folds;
std::vector<int> seeds;
};
class GridSearch {
public:
explicit GridSearch(struct ConfigGrid& config);
void go();
void save();
~GridSearch() = default;
private:
struct ConfigGrid config;
};
} /* namespace platform */
#endif /* GRIDSEARCH_H */