from dataclasses import dataclass

class ModelConfig:
    num_agents: int = 10
    embedding_dim: int = 64
    input_dim: int = 1
    output_dim: int = 1
    simulation_type:str = "consensus"

class TrainConfig:
    epochs: float = 100
    learning_rate: float = 1e-3
    verbose: bool = True
    log: bool = True
    log_epoch_interval: int = 10