include config for Tangos

Author: AnFreTh

mambular/configs/__init__.py

@@ -14,8 +14,10 @@
 from .trompt_config import DefaultTromptConfig
 from .base_config import BaseConfig
 from .enode_config import DefaultENODEConfig
+from .tangos_config import DefaultTangosConfig
 
 __all__ = [
+    "DefaultTangosConfig",
     "DefaultENODEConfig",
     "DefaultTromptConfig",
     "DefaultAutoIntConfig",

mambular/configs/tangos_config.py

@@ -0,0 +1,37 @@
+from collections.abc import Callable
+from dataclasses import dataclass, field
+import torch.nn as nn
+from .base_config import BaseConfig
+
+
+@dataclass
+class DefaultTangosConfig(BaseConfig):
+    """Configuration class for the default Multi-Layer Perceptron (TANGOS) model with predefined hyperparameters.
+
+    Parameters
+    ----------
+    layer_sizes : list, default=(256, 128, 32)
+        Sizes of the layers in the TANGOS.
+    activation : callable, default=nn.ReLU()
+        Activation function for the TANGOS layers.
+    skip_layers : bool, default=False
+        Whether to skip layers in the TANGOS.
+    dropout : float, default=0.2
+        Dropout rate for regularization.
+    use_glu : bool, default=False
+        Whether to use Gated Linear Units (GLU) in the TANGOS.
+    skip_connections : bool, default=False
+        Whether to use skip connections in the TANGOS.
+    """
+
+    # Architecture Parameters
+    layer_sizes: list = field(default_factory=lambda: [256, 128, 32])
+    activation: Callable = nn.ReLU()  # noqa: RUF009
+    skip_layers: bool = False
+    dropout: float = 0.2
+    use_glu: bool = False
+    skip_connections: bool = False
+    lamda1: float = 0.5
+    lamda2: float = 0.1
+    subsample: float = 0.5
+