Skip to content

Conversation

@jackjii79
Copy link
Contributor

@jackjii79 jackjii79 commented Jan 2, 2026

https://github.com/h2oai/h2oai/issues/34827

Due to the complexity of TabPFN, automation regression is skipped, manual testing results show down below

Copilot AI review requested due to automatic review settings January 2, 2026 22:00
Copy link

Copilot AI left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Pull request overview

This PR adds new TabPFN (Prior-data Fitted Networks) model functionality to Driverless AI, introducing both an outlier detection transformer and a supervised model for classification and regression tasks. The implementation includes GPU-accelerated inference, SHAP-based explanations, and support for datasets with up to 10,000 samples.

Key changes:

  • Implements TabPFN-based unsupervised outlier detection with density estimation via permutation sampling
  • Adds TabPFN supervised model with support for regression, binary, and multi-class classification (including many-class scenarios via codebook-based decomposition)
  • Includes SAGE/Shapley value computation for global feature importance and local explanations

Reviewed changes

Copilot reviewed 1 out of 1 changed files in this pull request and generated 20 comments.

File Description
transformers/outliers/tabpfn_outlier.py Implements TabPFN-based outlier detection transformer that calculates negative log probability scores using conditional probability estimation across feature permutations
models/algorithms/tabpfn_model.py Implements TabPFN supervised model with regression/classification support, many-class handling via codebook decomposition, and SHAP-based explanations using SAGE simulation

💡 Add Copilot custom instructions for smarter, more guided reviews. Learn how to get started.

@jackjii79 jackjii79 self-assigned this Jan 2, 2026
@jackjii79
Copy link
Contributor Author

Testing results:

  • TabPFNModel
    ✅ Classification
    ✅ Regression
Screenshot 2026-01-05 at 8 57 30 AM

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant