41 lines
1.1 KiB
Python
41 lines
1.1 KiB
Python
"""
|
|
Commander AI Lab - Training Subpackage
|
|
==========================================
|
|
Provides RL training infrastructure including PPO, rollout buffers,
|
|
Forge episode generation, self-play, and distillation.
|
|
"""
|
|
|
|
from ml.training.rollout_buffer import RolloutBuffer
|
|
from ml.training.reward import compute_intermediate_reward, compute_terminal_reward, RewardConfig
|
|
from ml.training.decision_exporter import DecisionExporter
|
|
from ml.training.forge_episode_generator import (
|
|
ForgeTransition,
|
|
ForgeEpisode,
|
|
ForgeEpisodeQueue,
|
|
forge_producer,
|
|
ppo_consumer,
|
|
run_forge_pipeline,
|
|
parse_forge_jsonl,
|
|
snapshots_to_episode,
|
|
)
|
|
|
|
__all__ = [
|
|
# Rollout & Buffers
|
|
"RolloutBuffer",
|
|
# Rewards
|
|
"compute_intermediate_reward",
|
|
"compute_terminal_reward",
|
|
"RewardConfig",
|
|
# Decision export
|
|
"DecisionExporter",
|
|
# Forge episode generation (Phase 3)
|
|
"ForgeTransition",
|
|
"ForgeEpisode",
|
|
"ForgeEpisodeQueue",
|
|
"forge_producer",
|
|
"ppo_consumer",
|
|
"run_forge_pipeline",
|
|
"parse_forge_jsonl",
|
|
"snapshots_to_episode",
|
|
]
|