Skip to content

Commit 5703d3b

Browse files
committed
adjusting docstrings for true defaults
1 parent e6b90dc commit 5703d3b

2 files changed

Lines changed: 10 additions & 10 deletions

File tree

mambular/configs/mambular_config.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -75,9 +75,9 @@ class DefaultMambularConfig:
7575
Whether to shuffle the embeddings before being passed to the Mamba layers.
7676
layer_norm_eps : float, default=1e-05
7777
Epsilon value for layer normalization.
78-
AD_weight_decay : bool, default=False
78+
AD_weight_decay : bool, default=True
7979
whether weight decay is also applied to A-D matrices.
80-
BC_layer_norm: bool, default=True
80+
BC_layer_norm: bool, default=False
8181
whether to apply layer normalization to B-C matrices.
8282
cat_encoding : str, default="int"
8383
whether to use integer encoding or one-hot encoding for cat features.
@@ -116,6 +116,6 @@ class DefaultMambularConfig:
116116
use_cls: bool = False
117117
shuffle_embeddings: bool = False
118118
layer_norm_eps: float = 1e-05
119-
AD_weight_decay: bool = False
120-
BC_layer_norm: bool = True
119+
AD_weight_decay: bool = True
120+
BC_layer_norm: bool = False
121121
cat_encoding: str = "int"

mambular/models/mambular.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -81,9 +81,9 @@ class MambularRegressor(SklearnBaseRegressor):
8181
Whether to shuffle the embeddings before being passed to the Mamba layers.
8282
layer_norm_eps : float, default=1e-05
8383
Epsilon value for layer normalization.
84-
AD_weight_decay : bool, default=False
84+
AD_weight_decay : bool, default=True
8585
whether weight decay is also applied to A-D matrices.
86-
BC_layer_norm: bool, default=True
86+
BC_layer_norm: bool, default=False
8787
whether to apply layer normalization to B-C matrices.
8888
cat_encoding : str, default="int"
8989
whether to use integer encoding or one-hot encoding for cat features.
@@ -208,9 +208,9 @@ class MambularClassifier(SklearnBaseClassifier):
208208
Whether to shuffle the embeddings before being passed to the Mamba layers.
209209
layer_norm_eps : float, default=1e-05
210210
Epsilon value for layer normalization.
211-
AD_weight_decay : bool, default=False
211+
AD_weight_decay : bool, default=True
212212
whether weight decay is also applied to A-D matrices.
213-
BC_layer_norm: bool, default=True
213+
BC_layer_norm: bool, default=False
214214
whether to apply layer normalization to B-C matrices.
215215
cat_encoding : str, default="int"
216216
whether to use integer encoding or one-hot encoding for cat features.
@@ -338,9 +338,9 @@ class MambularLSS(SklearnBaseLSS):
338338
Whether to shuffle the embeddings before being passed to the Mamba layers.
339339
layer_norm_eps : float, default=1e-05
340340
Epsilon value for layer normalization.
341-
AD_weight_decay : bool, default=False
341+
AD_weight_decay : bool, default=True
342342
whether weight decay is also applied to A-D matrices.
343-
BC_layer_norm: bool, default=True
343+
BC_layer_norm: bool, default=False
344344
whether to apply layer normalization to B-C matrices.
345345
cat_encoding : str, default="int"
346346
whether to use integer encoding or one-hot encoding for cat features.

0 commit comments

Comments
 (0)