Ver Fonte

1. 修改了partial和full的图片名称

maxfeng há 1 mês atrás
pai
commit
30e54075f6

+ 1 - 1
Lib/future/MAPatternStrategy_v002.py

@@ -204,7 +204,7 @@ def initialize(context):
     
     # 策略品种选择策略配置
     # 方案1:全品种策略 - 考虑所有配置的期货品种
-    g.strategy_focus_symbols = []  # 空列表表示考虑所有品种
+    g.strategy_focus_symbols = ['AO']  # 空列表表示考虑所有品种
     
     # 方案2:精选品种策略 - 只交易流动性较好的特定品种(如需使用请取消下行注释)
     # g.strategy_focus_symbols = ['RM', 'CJ', 'CY', 'JD', 'L', 'LC', 'SF', 'SI']

+ 5 - 3
Lib/future/trading_training_tool.py

@@ -807,7 +807,7 @@ def main():
         random.shuffle(open_trades)
         selected_trade = random.choice(open_trades)
 
-    print(f"选中交易: {selected_trade['contract_code']} - {selected_trade['trade_date']} - {selected_trade['direction']}")
+    # print(f"选中交易: {selected_trade['contract_code']} - {selected_trade['trade_date']} - {selected_trade['direction']}")
     print(f"剩余未处理交易: {len(open_trades) - 1} 个")
 
     # 5. 获取K线数据
@@ -825,7 +825,8 @@ def main():
 
     # 6. 显示部分K线图
     print("\n=== 步骤6: 显示部分K线图 ===")
-    partial_image_path = os.path.join(output_dir, f"partial_{selected_trade['trade_pair_id']}.png")
+    partial_image_name = f"partial_{selected_trade['contract_code']}_{selected_trade['trade_date']}_{selected_trade['direction']}.png"
+    partial_image_path = os.path.join(output_dir, partial_image_name)
     plot_partial_kline(
         kline_data, trade_idx, selected_trade['trade_price'],
         selected_trade['direction'], selected_trade['contract_code'],
@@ -838,7 +839,8 @@ def main():
 
     # 8. 显示完整K线图
     print("\n=== 步骤7: 显示完整K线图 ===")
-    full_image_path = os.path.join(output_dir, f"full_{selected_trade['trade_pair_id']}.png")
+    full_image_name = f"full_{selected_trade['contract_code']}_{selected_trade['trade_date']}_{selected_trade['direction']}.png"
+    full_image_path = os.path.join(output_dir, full_image_name)
     plot_full_kline(
         kline_data, trade_idx, selected_trade['trade_price'],
         selected_trade['direction'], selected_trade['contract_code'],