rmontanana/Platform
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Change default smooth type in Experiment

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2024-06-10 15:50:54 +02:00
parent f800772149
commit 0f06f8971e
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/main/Experiment.h
View File

@@ -28,6 +28,7 @@ namespace platform {
Experiment& setSmoothSrategy(const std::string& smooth_strategy) Experiment& setSmoothSrategy(const std::string& smooth_strategy)
{ {
this->smooth_strategy = smooth_strategy; this->result.setSmoothStrategy(smooth_strategy); this->smooth_strategy = smooth_strategy; this->result.setSmoothStrategy(smooth_strategy);
std::cout << "Experiment: Smoothing strategy: [" << smooth_strategy << "]" << std::endl;
if (smooth_strategy == "OLD_LAPLACE") if (smooth_strategy == "OLD_LAPLACE")
smooth_type = bayesnet::Smoothing_t::OLD_LAPLACE; smooth_type = bayesnet::Smoothing_t::OLD_LAPLACE;
else if (smooth_strategy == "LAPLACE") else if (smooth_strategy == "LAPLACE")
@@ -38,6 +39,7 @@ namespace platform {
std::cerr << "Experiment: Unknown smoothing strategy: " << smooth_strategy << std::endl; std::cerr << "Experiment: Unknown smoothing strategy: " << smooth_strategy << std::endl;
exit(1); exit(1);
} }
std::cout << "Experiment: " << (smooth_type == bayesnet::Smoothing_t::CESTNIK) << " " << static_cast<int>(smooth_type) << std::endl;
return *this; return *this;
} }
Experiment& setLanguageVersion(const std::string& language_version) { this->result.setLanguageVersion(language_version); return *this; } Experiment& setLanguageVersion(const std::string& language_version) { this->result.setLanguageVersion(language_version); return *this; }
@@ -60,7 +62,7 @@ namespace platform {
std::vector<int> randomSeeds; std::vector<int> randomSeeds;
std::string discretization_algo; std::string discretization_algo;
std::string smooth_strategy; std::string smooth_strategy;
bayesnet::Smoothing_t smooth_type = bayesnet::Smoothing_t::OLD_LAPLACE; bayesnet::Smoothing_t smooth_type{ bayesnet::Smoothing_t::NONE };
HyperParameters hyperparameters; HyperParameters hyperparameters;
int nfolds{ 0 }; int nfolds{ 0 };
int max_name{ 7 }; // max length of dataset name for formatting (default 7) int max_name{ 7 }; // max length of dataset name for formatting (default 7)