"""机器学习训练配置示例"""

import hydra
from omegaconf import DictConfig
from rich.console import Console
from rich.tree import Tree

console = Console()

@hydra.main(version_base=None, config_path="conf", config_name="config")
def train(cfg: DictConfig) -> None:
    console.print("\n[bold green]🤖 机器学习训练配置示例[/bold green]\n")
    
    tree = Tree("🎯 Training Configuration")
    
    model_tree = tree.add("[yellow]Model")
    model_tree.add(f"Name: {cfg.model.name}")
    model_tree.add(f"Layers: {cfg.model.layers}")
    
    optimizer_tree = tree.add("[cyan]Optimizer")
    optimizer_tree.add(f"Name: {cfg.optimizer.name}")
    optimizer_tree.add(f"Learning Rate: {cfg.optimizer.lr}")
    
    dataset_tree = tree.add("[magenta]Dataset")
    dataset_tree.add(f"Name: {cfg.dataset.name}")
    dataset_tree.add(f"Classes: {cfg.dataset.num_classes}")
    
    console.print(tree)
    console.print("\n[bold green]✅ 配置加载完成！[/bold green]\n")

if __name__ == "__main__":
    train()