Skip to content

Tabnet and v1 ready

Compare
Choose a tag to compare
@jrzaurin jrzaurin released this 25 Jun 17:52
· 518 commits to master since this release
a71699d

This release represents a major step forward for the library in terms of functionalities and flexibility:

  1. Ported TabNet from the fantastic implementation of the guys at dreamquark-ai.
  2. Callbacks are now more flexible and save more information.
  3. The save method in the Trainer is more flexible and transparent
  4. The library has extensively been tested via experiments against LightGBM (see here)