Skip to content

tabnet 0.9.0

Latest

Choose a tag to compare

@cregouby cregouby released this 12 Jun 16:23

New features

  • add the augment() function and document the predict() function (#190).

Improvements and fixes

  • Ancestor matrix is now taken into account, making hierarchical classification now effective (#188).
  • Fix steady loss due to frozen optimizer after checkpointing to disk when running on "cuda" and "mps" accelerators.
  • multiple R files renamed to improve clarity.