load trained df pkl file with bz2

This commit is contained in:
longyu 2022-09-18 10:24:59 +02:00
parent ae79f79d92
commit f4bf5b4018

View File

@ -469,7 +469,7 @@ class FreqaiDataDrawer:
dk.label_list = dk.data["label_list"] dk.label_list = dk.data["label_list"]
dk.data_dictionary["train_features"] = pd.read_pickle( dk.data_dictionary["train_features"] = pd.read_pickle(
dk.data_path / f"{dk.model_filename}_trained_df.pkl" dk.data_path / f"{dk.model_filename}_trained_df.pkl.bz2"
) )
# try to access model in memory instead of loading object from disk to save time # try to access model in memory instead of loading object from disk to save time