fix: add output_dim as an argument

bactgraph/modeling/train.py

@@ -60,6 +60,7 @@ def __init__(self):
     test: bool = False
     input_dim: int = 480
     hidden_dim: int = 480
+    output_dim: int = 1
     num_layers: int = 3
     num_heads: int = 4
     dropout: float = 0.2