fix: allow classifiers to work

This commit is contained in:
robcaulk
2024-05-11 16:21:15 +02:00
parent 42705374d0
commit 2d069d6156

View File

@@ -286,8 +286,7 @@ class FreqaiDataDrawer:
new_pred["date_pred"] = dataframe["date"] new_pred["date_pred"] = dataframe["date"]
# set everything to nan except date_pred # set everything to nan except date_pred
columns_to_nan = new_pred.columns.difference(['date_pred', 'date']) columns_to_nan = new_pred.columns.difference(['date_pred', 'date'])
new_pred[columns_to_nan] = new_pred[columns_to_nan].astype( new_pred[columns_to_nan] = None
float).values * np.nan
hist_preds = self.historic_predictions[pair].copy() hist_preds = self.historic_predictions[pair].copy()