Refactor crossvalidation to remove unneeded params

2023-07-29 16:49:06 +02:00
parent adc0ca238f
commit c4f3e6f19a
3 changed files with 17 additions and 14 deletions
--- a/src/Platform/Experiment.h
+++ b/src/Platform/Experiment.h
@@ -105,8 +105,7 @@ namespace platform {
        Experiment& setDuration(float duration) { this->duration = duration; return *this; }
        string get_file_name();
        void save(string path);
-        //Result cross_validation(const string& path, const string& fileName);
-        Result cross_validation(string model_name, torch::Tensor& X, torch::Tensor& y, vector<string> features, string className, map<string, vector<int>> states);
+        Result cross_validation(const string& path, const string& fileName);
        void show();
    };
 }