20 lines
355 B
Python
20 lines
355 B
Python
|
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
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
|