parallel mode add to ReccurenceExtractor

aimclub · Aug 21, 2023 · fb88e99 · fb88e99
1 parent 634d7dc
commit fb88e99
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/fedot_ind/core/models/recurrence/RecurrenceExtractor.py b/fedot_ind/core/models/recurrence/RecurrenceExtractor.py
@@ -66,7 +66,7 @@ def generate_vector_from_ts(self, ts_frame: pd.DataFrame) -> pd.DataFrame:
         """
 
         parallel = Parallel(n_jobs=self.n_processes, verbose=0, pre_dispatch="2*n_jobs")
-        components_and_vectors = parallel(delayed(self.generate_features_from_ts)(sample) for sample in ts_frame.value)
+        components_and_vectors = parallel(delayed(self._ts_chunk_function)(component) for component in ts_frame)
         if self.image_mode:
             components_and_vectors = np.asarray(components_and_vectors)
             components_and_vectors = components_and_vectors[:, np.newaxis, :, :]