https://github.com/chrislemke/scikit-tabtrans

TabTransformer ready for Scikit learn 🧑‍🔬
https://github.com/chrislemke/scikit-tabtrans

deep-learning machine-learning python scikit-learn transformer

Last synced: 26 days ago
JSON representation

TabTransformer ready for Scikit learn 🧑‍🔬

• Host: GitHub
• URL: https://github.com/chrislemke/scikit-tabtrans
• Owner: chrislemke
• License: mit
• Created: 2022-09-16T07:42:20.000Z (about 2 years ago)
• Default Branch: main
• Last Pushed: 2023-02-11T02:06:35.000Z (over 1 year ago)
• Last Synced: 2024-04-25T12:20:49.628Z (6 months ago)
• Topics: deep-learning, machine-learning, python, scikit-learn, transformer
• Language: Python
• Homepage: https://pypi.org/project/scikit-tabtrans/
• Size: 681 KB
• Stars: 1
• Watchers: 2
• Forks: 0
• Open Issues: 1
• Metadata Files:
  • Readme: README.md
  • License: LICENSE.md
  • Security: SECURITY.md

Awesome Lists containing this project

README

        
![TabTransformer](https://github.com/chrislemke/scikit-TabTrans/blob/main/images/tab_transfomer.png)

# scikit-TabTrans

## TabTransformer ready for Scikit learn

The idea behind this small project is to provide the [TabTransformer](https://arxiv.org/pdf/2012.06678.pdf) as a

[Scikit-Learn](https://scikit-learn.org/stable/) model. So it can be e.g. used as a part of a [StackingClassifier](https://scikit-learn.org/stable/modules/generated/sklearn.ensemble.StackingClassifier.html). We use the [pytorch-widedeep](https://github.com/jrzaurin/pytorch-widedeep) implementation of the TabTransformer. Checkout [pytorch-widedeep](https://github.com/jrzaurin/pytorch-widedeep) it's a great project!