Skip to content

Commit 07164f5

Browse files
committed
adjusting configs
1 parent d413fd8 commit 07164f5

3 files changed

Lines changed: 3 additions & 0 deletions

File tree

mambular/configs/fttransformer_config.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -85,3 +85,4 @@ class DefaultFTTransformerConfig:
8585
layer_norm_eps: float = 1e-05
8686
transformer_dim_feedforward: int = 256
8787
numerical_embedding: str = "ple"
88+
cat_encoding: str = "int"

mambular/configs/mambular_config.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -116,3 +116,4 @@ class DefaultMambularConfig:
116116
layer_norm_eps: float = 1e-05
117117
AD_weight_decay: bool = False
118118
BC_layer_norm: bool = True
119+
cat_encoding: str = "int"

mambular/configs/tabtransformer_config.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -84,3 +84,4 @@ class DefaultTabTransformerConfig:
8484
transformer_activation: callable = ReGLU()
8585
layer_norm_eps: float = 1e-05
8686
transformer_dim_feedforward: int = 512
87+
cat_encoding: str = "int"

0 commit comments

Comments
 (0)