Add only parameter to gridsearch

src/Platform/GridSearch.cc

@@ -123,15 +123,21 @@ namespace platform {
                 }
             }
             catch (const std::exception& e) {
-                std::cerr << "Error loading previous results: " << e.what() << std::endl;
+                std::cerr << "* There were no previous results" << std::endl;
+                std::cerr << "* Initizalizing new results" << std::endl;
+                results = json();
             }
             // Remove datasets already processed
             vector< string >::iterator it = datasets_names.begin();
             while (it != datasets_names.end()) {
                 if (*it != config.continue_from) {
                     it = datasets_names.erase(it);
-                } else
-                    break;
+                } else {
+                    if (config.only)
+                        ++it;
+                    else
+                        break;
+                }
             }
         }
         // Create model

src/Platform/GridSearch.h

@@ -17,6 +17,7 @@ namespace platform {
         std::string output_file;
         std::string continue_from;
         bool quiet;
+        bool only; // used with continue_from to only compute that dataset
         bool discretize;
         bool stratified;
         int n_folds;

src/Platform/b_grid.cc

@@ -23,9 +23,10 @@ argparse::ArgumentParser manageArguments(std::string program_name)
             }
     );
     program.add_argument("--discretize").help("Discretize input datasets").default_value((bool)stoi(env.get("discretize"))).implicit_value(true);
+    program.add_argument("--stratified").help("If Stratified KFold is to be done").default_value((bool)stoi(env.get("stratified"))).implicit_value(true);
     program.add_argument("--quiet").help("Don't display detailed progress").default_value(false).implicit_value(true);
     program.add_argument("--continue").help("Continue computing from that dataset").default_value("No");
-    program.add_argument("--stratified").help("If Stratified KFold is to be done").default_value((bool)stoi(env.get("stratified"))).implicit_value(true);
+    program.add_argument("--only").help("Used with continue to compute that dataset only").default_value(false).implicit_value(true);
     program.add_argument("--score").help("Score used in gridsearch").default_value("accuracy");
     program.add_argument("-f", "--folds").help("Number of folds").default_value(stoi(env.get("n_folds"))).scan<'i', int>().action([](const std::string& value) {
         try {
@@ -58,8 +59,12 @@ int main(int argc, char** argv)
         config.stratified = program.get<bool>("stratified");
         config.n_folds = program.get<int>("folds");
         config.quiet = program.get<bool>("quiet");
+        config.only = program.get<bool>("only");
         config.seeds = program.get<std::vector<int>>("seeds");
         config.continue_from = program.get<std::string>("continue");
+        if (config.continue_from == "No" && config.only) {
+            throw std::runtime_error("Cannot use --only without --continue");
+        }
     }
     catch (const exception& err) {
         cerr << err.what() << std::endl;