Skip to content

Add NPE-PFN #1682

@jsvetter

Description

@jsvetter

🚀 Feature Request

Add NPE-PFN to the library (in the same way as autogluon)

NPE-PFN is easy-to-use and has strong performance, particularly for small simulation budgets. But currently it is based on TabPFNv2, which uses a special license. AFAIK, the initial decision was to solve this via an external repo.

However, I recently dabbled with autogluon (developed by Amazon), which also uses the Apache 2.0 license. To conform to the special prior labs license, they simply log once (via a global variable) the necessary attribution line. See the code here
The same could be done in the SBI library.

What do you think?

Describe alternatives you've considered

Don't add NPE-PFN (i.e., just keep it in an external repo), or use a different tabular foundation model.

Metadata

Metadata

Assignees

No one assigned

    Labels

    enhancementNew feature or request

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions