chore: add TODO to save the predictions

aimclub · Sep 10, 2024 · 77e47ca · 77e47ca
1 parent ef90a68
commit 77e47ca
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/fedot/core/optimisers/objective/data_objective_eval.py b/fedot/core/optimisers/objective/data_objective_eval.py
@@ -49,6 +49,7 @@ def __init__(self,
         self._validation_blocks = validation_blocks
         self._pipelines_cache = pipelines_cache
         self._preprocessing_cache = preprocessing_cache
+        # TODO: self._data_cache = data_cache
         self._log = default_log(self)
         self._do_unfit = do_unfit
 
@@ -111,7 +112,7 @@ def prepare_graph(self, graph: Pipeline, train_data: InputData,
 
         # load preprocessing
         graph.try_load_from_cache(self._pipelines_cache, self._preprocessing_cache, fold_id)
-        graph.fit(
+        predicted_train = graph.fit(
             train_data,
             n_jobs=n_jobs,
             time_constraint=self._time_constraint
@@ -121,6 +122,9 @@ def prepare_graph(self, graph: Pipeline, train_data: InputData,
             self._pipelines_cache.save_pipeline(graph, fold_id)
         if self._preprocessing_cache is not None:
             self._preprocessing_cache.add_preprocessor(graph, fold_id)
+        # TODO: 
+        # if self._data_cache is not None:
+            # self._data_cache.save_data(graph, predicted_train, fold_id)
 
         return graph