This commit is contained in:
zhangkun9038@dingtalk.com 2025-04-29 14:20:23 +08:00
parent 1eedae044a
commit 3a721996a0

View File

@ -338,7 +338,8 @@ class FreqaiExampleStrategy(IStrategy):
dataframe["sell_rsi_pred"] = dataframe["sell_rsi_pred"].fillna(dataframe["sell_rsi_pred"].median())
# 检查并处理 NaN 值
for col in pred_columns + ["&-sell_rsi", "&-stoploss", "&-roi_0", "buy_rsi_pred", "sell_rsi_pred"]:
all_columns = list(pred_columns.keys()) + ["&-sell_rsi", "&-stoploss", "&-roi_0", "buy_rsi_pred", "sell_rsi_pred"]
for col in all_columns:
if col in dataframe.columns:
if dataframe[col].isna().any():
logger.warning(f"{col} 包含 NaN填充为默认值")