Merge branch 'save_df_as_fp32' into strategies

This commit is contained in:
longyu 2022-09-18 10:51:06 +02:00
commit 1f80cc9227

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"
) )