ensure data kitchen thread count is propagated to pipeline

This commit is contained in:
robcaulk
2023-06-08 12:33:08 +02:00
parent 88337b6c5e
commit 33b028b104
7 changed files with 12 additions and 32 deletions

View File

@@ -118,8 +118,8 @@ class BaseReinforcementLearningModel(IFreqaiModel):
# normalize all data based on train_dataset only
prices_train, prices_test = self.build_ohlc_price_dataframes(dk.data_dictionary, pair, dk)
dk.feature_pipeline = self.define_data_pipeline()
dk.label_pipeline = self.define_label_pipeline()
dk.feature_pipeline = self.define_data_pipeline(threads=dk.thread_count)
dk.label_pipeline = self.define_label_pipeline(threads=dk.thread_count)
(dd["train_features"],
dd["train_labels"],