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,7 +981,9 @@ class FreqaiDataKitchen:
append_df[label] = predictions[label]
if append_df[label].dtype == object:
continue
if "labels_mean" in self.data:
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