Skip to content

v1.5.1

Choose a tag to compare

@dholzmueller dholzmueller released this 25 Jun 21:24
· 25 commits to main since this release
  • v1.5.1: fixed a device bug in TabM for GPU
  • v1.5.0:
    • added n_repeats parameter to scikit-learn interfaces for repeated cross-validation
    • HPO sklearn interfaces (the ones using random search)
      can now do weighted ensembling instead by setting use_caruana_ensembling=True.
      Removed the RealMLP_Ensemble_Classifier and RealMLP_Ensemble_Regressor from v1.4.2
      since they are now redundant through this feature.
    • renamed space parameter of GBDT HPO interface
      to hpo_space_name so now it also works with non-TPE versions.
    • Added new TabArena search spaces for boosted trees (not TPE),
      which should be almost equivalent to the ones from TabArena
      except for the early stopping logic.
    • TabM now supports val_metric_name for early stopping on different metrics.
    • fixed issues #20 and #21 regarding HPO
    • small updates for the "Rethinking Early Stopping" paper