search

ersilia-os / chempfn

Ensemble-based, size-agnostic wrapper for the TabPFN classifier
GNU General Public License v3.0
28 stars 0 forks source link

Early stopping and other fixes #31

Closed DhanshreeA closed 1 year ago

DhanshreeA commented 1 year ago
  1. Implement early stopping at predict time using early_stopping_rounds and tolerance which checks if predictions for any samples have changed within the tolerance threshold, and if predictions for such samples do not change for early_stopping_rounds, then those samples are frozen and not passed to TabPFN predict.
  2. Implements model load and save functionality to save generated ensembles and load the same ensembles in a new instance of EnsembleTabPFN. Currently only the pickle format is supported.
  3. Major refactoring involving abstraction of ensemble building from EnsembleTabPFN (similar to data loading functionalities provided in other ML frameworks).
  4. Added tests for EnsembleTabPFN creation and model load and save functionality.
DhanshreeA commented 1 year ago

Addresses https://github.com/ersilia-os/ensemble-tabpfn/issues/29 Addresses https://github.com/ersilia-os/ensemble-tabpfn/issues/28 Addresses https://github.com/ersilia-os/ensemble-tabpfn/issues/26