61 lines
1.9 KiB
Python
61 lines
1.9 KiB
Python
"""
|
||
train_ftmo_island.py — FTMO 专属 + Island 多起点并行训练
|
||
|
||
训练 index 组，使用 IslandAlphaEngine：
|
||
- 3 个独立模型（islands）
|
||
- 每 200 步迁移一次 elite
|
||
- 自适应噪声、部分参数重置、elite decay 已启用
|
||
"""
|
||
import sys
|
||
import time
|
||
from pathlib import Path
|
||
|
||
sys.path.insert(0, str(Path(__file__).parent))
|
||
|
||
from config import Config
|
||
from data_pipeline.data_manager import MT5DataManager
|
||
from data_pipeline.fetcher import MT5DataFetcher
|
||
from model_core.config import ModelConfig
|
||
from model_core.island_engine import IslandAlphaEngine
|
||
|
||
|
||
def main():
|
||
offline = "--offline" in sys.argv
|
||
t0 = time.time()
|
||
|
||
print(f"\n{'='*60}")
|
||
print(f" FTMO + Island 多起点并行训练 — index 组")
|
||
print(f"{'='*60}")
|
||
print(f" 奖励模式: {ModelConfig.REWARD_MODE}")
|
||
print(f" 公式长度: {ModelConfig.MAX_FORMULA_LEN}")
|
||
print(f" Islands : {ModelConfig.N_ISLANDS}")
|
||
print(f" Migration: every {ModelConfig.MIGRATION_INTERVAL} steps, Top-{ModelConfig.MIGRATION_TOP_K}")
|
||
print(f" Adaptive noise: {ModelConfig.ADAPTIVE_NOISE}")
|
||
print(f" Partial reset: {ModelConfig.PARTIAL_RESET}")
|
||
print(f" Elite decay: {ModelConfig.ELITE_DECAY}")
|
||
print(f" 品种: {Config.SYMBOL_GROUPS['index']}")
|
||
print(f"{'='*60}\n")
|
||
|
||
with MT5DataFetcher(offline=offline) as fetcher:
|
||
# 临时切换 SYMBOLS 到 index 组
|
||
original_symbols = Config.SYMBOLS[:]
|
||
Config.SYMBOLS = Config.SYMBOL_GROUPS["index"]
|
||
try:
|
||
mgr = MT5DataManager(fetcher)
|
||
mgr.load()
|
||
engine = IslandAlphaEngine(mgr)
|
||
engine.train()
|
||
|
||
formula, score = engine.get_global_best()
|
||
print(f"\n最终全局最优: score={score:.4f}")
|
||
print(f"公式: {formula}")
|
||
finally:
|
||
Config.SYMBOLS = original_symbols
|
||
|
||
elapsed = time.time() - t0
|
||
print(f"\n总耗时 {elapsed/3600:.2f}h")
|
||
|
||
|
||
if __name__ == "__main__":
|
||
main()
|