Skip to content

Instantly share code, notes, and snippets.

@jrzaurin
Last active June 13, 2021 09:17
Show Gist options
  • Save jrzaurin/413a06862b240ea483d3e04a86a968dc to your computer and use it in GitHub Desktop.
Save jrzaurin/413a06862b240ea483d3e04a86a968dc to your computer and use it in GitHub Desktop.
LightGBM vs DL for the Bank Marketing dataset
model f1 auc runtime best_epoch_or_ntrees
tabresnet 0.4300 0.6501 92.52 11.0
tabtransformer 0.4200 0.6440 31.69 4.0
tabmlp 0.3856 0.6281 9.57 7.0
lightgbm 0.3852 0.6265 0.46 57.0
tabnet 0.3087 0.5943 77.88 13.0
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment