Skip to content

logit-per-default-for-tabpfn#460

Merged
mmschlk merged 5 commits intomainfrom
logit-per-default-for-tabpfn
Oct 29, 2025
Merged

logit-per-default-for-tabpfn#460
mmschlk merged 5 commits intomainfrom
logit-per-default-for-tabpfn

Conversation

@mmschlk
Copy link
Copy Markdown
Owner

@mmschlk mmschlk commented Oct 29, 2025

Motivation and Context

Closes #426


Public API Changes

  • No Public API changes
  • Yes, Public API changes (Details below)

How Has This Been Tested?

New test case and locally testing.


Checklist

  • The changes have been tested locally.
  • Documentation has been updated (if the public API or usage changes).
  • An entry has been added to CHANGELOG.md (if relevant for users).
  • The code follows the project's style guidelines.
  • I have considered the impact of these changes on the public API.

@mmschlk mmschlk moved this to 👀 In review in shapiq development Oct 29, 2025
@mmschlk mmschlk requested a review from Copilot October 29, 2025 16:28
@mmschlk mmschlk added this to the 1.4.0 milestone Oct 29, 2025
Copy link
Copy Markdown
Contributor

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 upgrades TabPFN support and implements a new prediction method priority system. The key changes include updating TabPFN from version 2.0.7 to 2.2.1 (via minimum version 2.1.3), introducing a new predict_logits method that is now prioritized over predict_proba for sklearn-like classifiers, and adding tests to verify the correct prediction function is used for TabPFN models.

  • Updated TabPFN dependency from >=2.0.7 to >=2.1.3 to support models with predict_logits method
  • Introduced predict_logits function and prioritized it over predict_proba in prediction function selection
  • Added new test fixtures and test cases to verify TabPFN models use logits-based predictions

Reviewed Changes

Copilot reviewed 5 out of 6 changed files in this pull request and generated 2 comments.

Show a summary per file
File Description
uv.lock Updates dependency versions including TabPFN (2.0.7 → 2.2.1), filelock, platformdirs, and adds new dependencies for TabPFN (pydantic, posthog, etc.)
pyproject.toml Updates minimum TabPFN version requirement from 2.0.7 to 2.1.3 and clarifies a comment about lightgbm
src/shapiq/explainer/utils.py Adds predict_logits function and prioritizes it over predict_proba in the prediction function selection logic
tests/shapiq/fixtures/models.py Adds two new fixtures (tabpfn_reg_model and tabpfn_clf_model) for testing TabPFN models
tests/shapiq/tests_unit/tests_explainer/test_prediction_function.py Adds new test file to verify that TabPFN classifiers use the predict_logits method
CHANGELOG.md Documents the new feature and clarifies the comment about lightgbm installation issues

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

Comment thread src/shapiq/explainer/utils.py
Comment thread CHANGELOG.md Outdated
@codecov
Copy link
Copy Markdown

codecov Bot commented Oct 29, 2025

Codecov Report

✅ All modified and coverable lines are covered by tests.

📢 Thoughts on this report? Let us know!

Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>
@mmschlk mmschlk merged commit 68678c0 into main Oct 29, 2025
10 checks passed
@github-project-automation github-project-automation Bot moved this from 👀 In review to ✅ Done in shapiq development Oct 29, 2025
@mmschlk mmschlk deleted the logit-per-default-for-tabpfn branch October 29, 2025 16:48
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

Status: ✅ Done

Development

Successfully merging this pull request may close these issues.

Use logits per default when explaining models like tabpfn.

2 participants