Skip to content

v0.4.8: WideDeep with the TabTransformer

Compare
Choose a tag to compare
@jrzaurin jrzaurin released this 11 Feb 13:22
· 520 commits to master since this release
f430864

This release represents an almost-complete refactor of the previous version and I consider the code in this version well tested and production-ready. The main reason why this release is not v1 is because I want to use it with a few more datasets, but at the same time I want the version to be public to see if others use it. Also, I want the changes from the last Beta and v1 to be not too significant.

This version is not backwards compatible (at all).

These are some of the structural changes:

  • Building of the model and training the model and now completely decoupled
  • Added the TabTransformer as a potential deeptabular component
  • Renamed many of the parameters so that they are consistent between models
  • Added the possibility of customising almost every single component: model component, losses, metrics and callbacks
  • Added R2 metrics for regression problems