search

automl / PFNs

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

PFN classifier #1

Closed omarswelam closed 5 months ago

omarswelam commented 7 months 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 5 months ago

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