--- a +++ b/src/ml_models/bilstm/config.py @@ -0,0 +1,34 @@ +# Base Dependencies +# ----------------- +from dataclasses import dataclass + + +# Configuration Classes +# --------------------- +@dataclass +class LSTMConfig: + hidden_size: int = 128 + emb_size: int = 256 # input size + num_layers: int = 2 + dropout: float = 0.25 + emb_droput: float = 0.25 + bidirectional: bool = True + bias: bool = True + batch_first: bool = True + + +@dataclass +class RDEmbeddingConfig: + input_dim: int = 5 + embedding_dim: int = 5 + scale: bool = False + freeze: bool = False + + +@dataclass +class EmbeddingConfig: + embedding_dim: int = 5 + vocab_size: int = 2 + padding_idx: int = 0 + freeze: bool = True + emb_path: str = ""