verify mean and std exists in model metadata

This commit is contained in:
Wagner Costa Santos 2022-11-03 13:27:56 -03:00
parent 6606a0113f
commit 356d79b38a

View File

@ -981,8 +981,10 @@ class FreqaiDataKitchen:
append_df[label] = predictions[label] append_df[label] = predictions[label]
if append_df[label].dtype == object: if append_df[label].dtype == object:
continue continue
append_df[f"{label}_mean"] = self.data["labels_mean"][label] if "labels_mean" in self.data:
append_df[f"{label}_std"] = self.data["labels_std"][label] append_df[f"{label}_mean"] = self.data["labels_mean"][label]
if "labels_std" in self.data:
append_df[f"{label}_std"] = self.data["labels_std"][label]
append_df["do_predict"] = do_predict append_df["do_predict"] = do_predict
if self.freqai_config["feature_parameters"].get("DI_threshold", 0) > 0: if self.freqai_config["feature_parameters"].get("DI_threshold", 0) > 0: