Skip to content

Instantly share code, notes, and snippets.

@jrzaurin
Last active June 13, 2021 09:14
Show Gist options
  • Save jrzaurin/97a0f9b92619865f355b9d03bd9071dd to your computer and use it in GitHub Desktop.
Save jrzaurin/97a0f9b92619865f355b9d03bd9071dd to your computer and use it in GitHub Desktop.
LightGBM vs DL for the Adult Census dataset
model acc runtime best_epoch_or_ntrees
lightgbm 0.8782 0.92 408.0
tabmlp 0.8722 205.36 62.0
tabtransformer 0.8718 288.64 32.0
tabnet 0.8704 422.30 26.0
tabresnet 0.8698 388.93 25.0
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment