search

automl / PFNs

Our maintained PFN repository. Come here to train SOTA PFNs.
Apache License 2.0
51 stars 10 forks source link

PFN classifier #1

Closed omarswelam closed 10 months ago

omarswelam commented 1 year ago

Adding PFNClassifier with multivalued classification functionality using a TransformerModel checkpoint under models_diff. Additionally, openml dataset generation scripts were added for testing. Implementations of added scripts follow the implementation procedures for the TabPFN classification class.

SamuelGabriel commented 10 months ago

this looks great, thank you. i am merging this now :)