Skip to content

tabensemb v0.2 release

Choose a tag to compare

View all tags
@LuoXueling LuoXueling released this 25 Sep 16:33
· 21 commits to main since this release
v0.2
9deb145
This commit was created on GitHub.com and signed with GitHub’s verified signature.
GPG key ID: B5690EEEBB952194
Verified
Learn about vigilant mode.

A framework to evaluate various models for tabular regression and classification tasks. The package integrates 25 machine learning (including deep learning) models for tabular prediction tasks from three well-established model bases:

  • autogluon
    • "LightGBM", "CatBoost", "XGBoost", "Random Forest", "Extremely Randomized Trees", "K-Nearest Neighbors", "Linear Regression", "Neural Network with MXNet", "Neural Network with PyTorch", "Neural Network with FastAI".
  • pytorch_widedeep
    • "TabMlp", "TabResnet", "TabTransformer", "TabNet", "SAINT", "ContextAttentionMLP", "SelfAttentionMLP", "FTTransformer", "TabPerceiver", "TabFastFormer".
  • pytorch_tabular
    • "Category Embedding", "NODE", "TabNet", "TabTransformer", "AutoInt", "FTTransformer".
Assets 2
Loading