fix: pandas.
This commit is contained in:
@@ -285,7 +285,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'])
|
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] = new_pred[columns_to_nan].astype(
|
||||||
float).values * np.nan
|
float).values * np.nan
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user