adelorenzo/stable
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge branch 'save_df_as_fp32' into strategies

Browse Source
This commit is contained in:
longyu
2022-09-18 10:33:14 +02:00
parent ff285fffe1 f4bf5b4018
commit 7348a98332
19 changed files with 778 additions and 125 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
freqtrade/freqai/data_drawer.py
View File

@@ -420,8 +420,8 @@ class FreqaiDataDrawer:
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
dk.data_dictionary["train_features"].to_pickle(
save_path / f"{dk.model_filename}_trained_df.pkl"
dk.data_dictionary["train_features"].astype("floa32").to_pickle(
save_path / f"{dk.model_filename}_trained_df.pkl.bz2"
)
dk.data_dictionary["train_dates"].to_pickle(
@@ -469,7 +469,7 @@ class FreqaiDataDrawer:
dk.label_list = dk.data["label_list"]
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