From 5703d3b473b65ba24e48dbaa444f8f13febb95b8 Mon Sep 17 00:00:00 2001 From: AnFreTh Date: Mon, 5 Aug 2024 10:15:33 +0000 Subject: [PATCH] adjusting docstrings for true defaults --- mambular/configs/mambular_config.py | 8 ++++---- mambular/models/mambular.py | 12 ++++++------ 2 files changed, 10 insertions(+), 10 deletions(-) diff --git a/mambular/configs/mambular_config.py b/mambular/configs/mambular_config.py index 4f3e495..2083961 100644 --- a/mambular/configs/mambular_config.py +++ b/mambular/configs/mambular_config.py @@ -75,9 +75,9 @@ class DefaultMambularConfig: Whether to shuffle the embeddings before being passed to the Mamba layers. layer_norm_eps : float, default=1e-05 Epsilon value for layer normalization. - AD_weight_decay : bool, default=False + AD_weight_decay : bool, default=True whether weight decay is also applied to A-D matrices. - BC_layer_norm: bool, default=True + BC_layer_norm: bool, default=False whether to apply layer normalization to B-C matrices. cat_encoding : str, default="int" whether to use integer encoding or one-hot encoding for cat features. @@ -116,6 +116,6 @@ class DefaultMambularConfig: use_cls: bool = False shuffle_embeddings: bool = False layer_norm_eps: float = 1e-05 - AD_weight_decay: bool = False - BC_layer_norm: bool = True + AD_weight_decay: bool = True + BC_layer_norm: bool = False cat_encoding: str = "int" diff --git a/mambular/models/mambular.py b/mambular/models/mambular.py index 09a518a..104448a 100644 --- a/mambular/models/mambular.py +++ b/mambular/models/mambular.py @@ -81,9 +81,9 @@ class MambularRegressor(SklearnBaseRegressor): Whether to shuffle the embeddings before being passed to the Mamba layers. layer_norm_eps : float, default=1e-05 Epsilon value for layer normalization. - AD_weight_decay : bool, default=False + AD_weight_decay : bool, default=True whether weight decay is also applied to A-D matrices. - BC_layer_norm: bool, default=True + BC_layer_norm: bool, default=False whether to apply layer normalization to B-C matrices. cat_encoding : str, default="int" whether to use integer encoding or one-hot encoding for cat features. @@ -208,9 +208,9 @@ class MambularClassifier(SklearnBaseClassifier): Whether to shuffle the embeddings before being passed to the Mamba layers. layer_norm_eps : float, default=1e-05 Epsilon value for layer normalization. - AD_weight_decay : bool, default=False + AD_weight_decay : bool, default=True whether weight decay is also applied to A-D matrices. - BC_layer_norm: bool, default=True + BC_layer_norm: bool, default=False whether to apply layer normalization to B-C matrices. cat_encoding : str, default="int" whether to use integer encoding or one-hot encoding for cat features. @@ -338,9 +338,9 @@ class MambularLSS(SklearnBaseLSS): Whether to shuffle the embeddings before being passed to the Mamba layers. layer_norm_eps : float, default=1e-05 Epsilon value for layer normalization. - AD_weight_decay : bool, default=False + AD_weight_decay : bool, default=True whether weight decay is also applied to A-D matrices. - BC_layer_norm: bool, default=True + BC_layer_norm: bool, default=False whether to apply layer normalization to B-C matrices. cat_encoding : str, default="int" whether to use integer encoding or one-hot encoding for cat features.