Build gridsearch structure
This commit is contained in:
30
src/Platform/GridSearch.h
Normal file
30
src/Platform/GridSearch.h
Normal 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 */
|
Reference in New Issue
Block a user