Add discretization algo to result

2024-06-06 18:33:01 +02:00
parent c4f4e332f6
commit 2202a81782
3 changed files with 8 additions and 0 deletions
--- a/src/commands/b_main.cpp
+++ b/src/commands/b_main.cpp
@@ -177,7 +177,9 @@ int main(int argc, char** argv)
     */
    auto env = platform::DotEnv();
    auto experiment = platform::Experiment();
+    std::string discretiz_algo = env.get("discretiz_algo");
    experiment.setTitle(title).setLanguage("c++").setLanguageVersion("13.2.1");
+    experiment.setDiscretizationAlgorithm(discretiz_algo);
    experiment.setDiscretized(discretize_dataset).setModel(model_name).setPlatform(env.get("platform"));
    experiment.setStratified(stratified).setNFolds(n_folds).setScoreName("accuracy");
    experiment.setHyperparameters(test_hyperparams);