Skip to content

Commit

Permalink
Update example.yaml
Browse files Browse the repository at this point in the history
  • Loading branch information
dev-geof authored May 30, 2024
1 parent 2c49a13 commit dc7fe0e
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions config/example.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -108,8 +108,8 @@ transformer_classification_parameters: # Hyperparameters for the transformer cla
nepochs: 500 # [int] Number of training epochs.
learning_rate: 0.001 # [float] Learning rate.
verbose: 1 # [int] Displayed information during training.
embedding: True # NO YET AVAILABLE - [bool] Include embedding layer in case of categorical input variables.
embedding_dim: 512 # NO YET AVAILABLE - [int] embedding layer output dimension
embedding: True # [bool] Include embedding layer in case of categorical input variables.
embedding_dim: 512 # [int] embedding layer output dimension

######################################
# validation configuration
Expand Down

0 comments on commit dc7fe0e

Please sign in to comment.