Skip to content

Instantly share code, notes, and snippets.

@jrzaurin
Last active June 13, 2021 09:18
Show Gist options
  • Save jrzaurin/1297adbb484ff8a8ba4072b289d6a0e8 to your computer and use it in GitHub Desktop.
Save jrzaurin/1297adbb484ff8a8ba4072b289d6a0e8 to your computer and use it in GitHub Desktop.
Results for the NYC Taxi rode duration dataset with TabMlp
mlp_hidden_dims mlp_activation mlp_dropout mlp_batchnorm mlp_batchnorm_last mlp_linear_first embed_dropout lr batch_size weight_decay optimizer lr_scheduler base_lr max_lr div_factor final_div_factor n_cycles val_loss_or_metric
auto relu 0.1 False False True 0.0 0.01 1024 0.0 Adam ReduceLROnPlateau 0.001 0.01 25 10000.0 5.0 79252.7786
auto relu 0.1 False False True 0.0 0.01 1024 0.0 AdamW ReduceLROnPlateau 0.001 0.01 25 10000.0 5.0 79440.6025
auto relu 0.1 False False False 0.1 0.01 1024 0.0 Adam ReduceLROnPlateau 0.001 0.01 25 10000.0 5.0 79477.5653
auto relu 0.1 False False False 0.1 0.01 1024 0.0 AdamW ReduceLROnPlateau 0.001 0.01 25 10000.0 5.0 79710.8550
auto relu 0.1 False False False 0.0 0.01 1024 0.0 AdamW ReduceLROnPlateau 0.001 0.01 25 10000.0 5.0 80214.7197
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment