Skip to content

Add Mitra-v2 classifier; enforce minimum AutoGluon version per foundation model - #293

Merged
shchur merged 1 commit into
masterfrom
mitra-v2-min-framework-version
Oct 7, 2026
Merged

shchur merged 1 commit into
masterfrom
mitra-v2-min-framework-version

Conversation

@shchur

@shchur shchur commented Oct 7, 2026

Copy link
Copy Markdown
Collaborator

Summary

  • Add mitra-v2-classifier (autogluon/mitra-classifier-2) to the tabular foundation models. Uses the stock AutoGluon MITRA model via hf_cls_model (drop-in weights for Mitra-v1). Regression (mitra-regressor-2) is not added yet: its 1,000-bin head isn't supported by stock AutoGluon.
  • Add min_framework_version to FoundationModelConfig (default 1.5; 1.6 for Toto-2.0, TabICLv2, TabDPT-Turbo, Nori, Mitra-v2 — the AG release that introduced them).
  • deploy() / predict() / predict_proba() raise a ValueError before launching any SageMaker job if framework_version is older than the model's minimum. Compared on major.minor; "latest" always passes. Skipped when custom_image_uri is set.

Testing

  • Unit tests for the Mitra-v2 hyperparameters and the version check (reject / accept / custom image skip).

@github-actions

github-actions Bot commented Oct 7, 2026

Copy link
Copy Markdown

Job PR-293-f7c0dc9 is done.
Docs are uploaded to https://d12sc05jpx1wj5.cloudfront.net/PR-293/f7c0dc9/index.html

@shchur
shchur merged commit aca1110 into master Oct 7, 2026
9 checks passed
@shchur
shchur deleted the mitra-v2-min-framework-version branch October 7, 2026 12:51
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