fix pickle file name

This commit is contained in:
robcaulk 2023-06-07 09:28:56 +02:00
parent 17d74429b5
commit f7f88aa14d

View File

@ -523,7 +523,7 @@ class FreqaiDataDrawer:
dk.data_dictionary["train_features"] = pd.read_pickle(
dk.data_path / f"{dk.model_filename}_trained_df.pkl"
)
with (dk.data_path / f"{dk.model_filename}_pipeline.pkl").open("rb") as fp:
with (dk.data_path / f"{dk.model_filename}_feature_pipeline.pkl").open("rb") as fp:
dk.feature_pipeline = cloudpickle.load(fp)
with (dk.data_path / f"{dk.model_filename}_label_pipeline.pkl").open("rb") as fp:
dk.label_pipeline = cloudpickle.load(fp)