Skip to content

Commit

Permalink
adjusting docstrings for true defaults
Browse files Browse the repository at this point in the history
  • Loading branch information
AnFreTh committed Aug 5, 2024
1 parent e6b90dc commit 5703d3b
Show file tree
Hide file tree
Showing 2 changed files with 10 additions and 10 deletions.
8 changes: 4 additions & 4 deletions mambular/configs/mambular_config.py
Original file line number Diff line number Diff line change
Expand Up @@ -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.
Expand Down Expand Up @@ -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"
12 changes: 6 additions & 6 deletions mambular/models/mambular.py
Original file line number Diff line number Diff line change
Expand Up @@ -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.
Expand Down Expand Up @@ -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.
Expand Down Expand Up @@ -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.
Expand Down

0 comments on commit 5703d3b

Please sign in to comment.