This commit is contained in:
longyu 2022-09-18 10:50:48 +02:00
parent f4bf5b4018
commit d6440fd74a

View File

@ -420,7 +420,7 @@ class FreqaiDataDrawer:
rapidjson.dump(dk.data, fp, default=self.np_encoder, number_mode=rapidjson.NM_NATIVE) rapidjson.dump(dk.data, fp, default=self.np_encoder, number_mode=rapidjson.NM_NATIVE)
# save the train data to file so we can check preds for area of applicability later # save the train data to file so we can check preds for area of applicability later
dk.data_dictionary["train_features"].astype("floa32").to_pickle( dk.data_dictionary["train_features"].astype("float32").to_pickle(
save_path / f"{dk.model_filename}_trained_df.pkl.bz2" save_path / f"{dk.model_filename}_trained_df.pkl.bz2"
) )