fix: for some reason we werent including high_price

This commit is contained in:
robcaulk
2024-04-14 14:19:12 +02:00
parent 65654a4a44
commit 7a401415e5

View File

@@ -675,6 +675,8 @@ class IFreqaiModel(ABC):
for return_str in dk.data['extra_returns_per_train']:
hist_preds_df[return_str] = dk.data['extra_returns_per_train'][return_str]
hist_preds_df['high_price'] = strat_df['high']
hist_preds_df['low_price'] = strat_df['low']
hist_preds_df['close_price'] = strat_df['close']
hist_preds_df['date_pred'] = strat_df['date']