trae解决分类目标无法获取问题

This commit is contained in:
zhangkun9038@dingtalk.com 2025-08-17 11:41:47 +08:00
parent 5e81d74070
commit febe57c744

View File

@ -1401,9 +1401,20 @@ class FreqaiPrimer(IStrategy):
try:
# 使用分类模型预测最优 first_length
# 检查所有可能的分类列
classification_cols = [col for col in dataframe.columns if "optimal_first_length" in str(col)]
logger.info(f"[{pair}] 📊 检测到分类相关列: {classification_cols}")
# 检查所有可能的分类列(精确匹配实际列名格式)
classification_cols = [col for col in dataframe.columns
if col in ["&*-optimal_first_length", "optimal_first_length_pred"]]
if classification_cols:
logger.info(f"[{pair}] 📊 检测到分类相关列: {classification_cols}")
else:
logger.debug(f"[{pair}] 📊 未检测到分类相关列")
# 调试:显示所有包含"optimal"的列名
all_optimal_cols = [col for col in dataframe.columns if "optimal" in str(col).lower()]
if all_optimal_cols:
logger.debug(f"[{pair}] 所有包含'optimal'的列: {all_optimal_cols}")
else:
logger.debug(f"[{pair}] 未找到任何包含'optimal'的列")
if "optimal_first_length_pred" in dataframe.columns:
predicted_value = dataframe["optimal_first_length_pred"].iloc[-1]
@ -1429,9 +1440,12 @@ class FreqaiPrimer(IStrategy):
logger.warning(f"[{pair}] ⚠️ &*-optimal_first_length 为NaN使用默认值: {first_length}")
else:
first_length = 2
logger.warning(f"[{pair}] ⚠️ 未找到任何分类模型列,使用默认值: {first_length}")
# 调试:显示所有列名
logger.debug(f"[{pair}] 可用列: {list(dataframe.columns)[:10]}...")
# 只在首次或特定条件下显示警告
if not hasattr(self, '_classification_warning_shown'):
logger.warning(f"[{pair}] ⚠️ 未找到分类模型列,使用默认值: {first_length}")
self._classification_warning_shown = True
else:
logger.debug(f"[{pair}] 未找到分类模型列,使用默认值: {first_length}")
# 根据 first_length 动态调整其他段长
second_length = first_length * 3