replicated mecc

train.py

@@ -0,0 +1,19 @@
+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
+    
+
+
+