The idea behind this small project is to provide the TabTransformer as a Scikit-Learn model. So it can be e.g. used as a part of a StackingClassifier. We use the pytorch-widedeep implementation of the TabTransformer. Checkout pytorch-widedeep it's a great project!