Skip to content

Support XGBoost 3.1#638

Merged
hcho3 merged 5 commits intodmlc:mainlinefrom
hcho3:support_xgboost_3.1
Nov 4, 2025
Merged

Support XGBoost 3.1#638
hcho3 merged 5 commits intodmlc:mainlinefrom
hcho3:support_xgboost_3.1

Conversation

@hcho3
Copy link
Collaborator

@hcho3 hcho3 commented Oct 31, 2025

Closes #635
Closes #633

TODOs

  • Support vector base_score field (multiple outputs)
  • Parse cats field. Note that we will parse the field but will not store them in the Treelite model. See [FEA] Store categorical encoding in Treelite model representation #639.
  • Update XGBoost integration tests to remove the use of legacy binary format, which was removed from XGBoost 3.1.
  • Add test coverage, to ensure that exceptions are thrown for XGBoost models with categorical encoding.

@hcho3 hcho3 self-assigned this Oct 31, 2025
@hcho3 hcho3 changed the title [WIP] Support XGBoost 3.1 Support XGBoost 3.1 Nov 4, 2025
@codecov
Copy link

codecov bot commented Nov 4, 2025

Codecov Report

❌ Patch coverage is 82.57576% with 23 lines in your changes missing coverage. Please review.
✅ Project coverage is 84.33%. Comparing base (aea9a94) to head (e233818).
⚠️ Report is 13 commits behind head on mainline.

Files with missing lines Patch % Lines
...el_loader/detail/xgboost_json/delegated_handler.cc 77.22% 23 Missing ⚠️
Additional details and impacted files
@@             Coverage Diff              @@
##           mainline     #638      +/-   ##
============================================
- Coverage     84.78%   84.33%   -0.45%     
============================================
  Files            77       77              
  Lines          6756     6869     +113     
  Branches        531      543      +12     
============================================
+ Hits           5728     5793      +65     
- Misses         1028     1076      +48     

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

🚀 New features to boost your workflow:
  • ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.

@hcho3 hcho3 merged commit 92977d4 into dmlc:mainline Nov 4, 2025
18 of 20 checks passed
@hcho3 hcho3 deleted the support_xgboost_3.1 branch November 4, 2025 06:57
rapids-bot bot pushed a commit to rapidsai/cuml that referenced this pull request Nov 14, 2025
Update Treelite to 4.6.1 to incorporate the following improvements:
* **Support XGBoost 3.1** (dmlc/treelite#638)
* Support scikit-learn 1.7 (dmlc/treelite#611)
* Fix FIL prediction for isolation forest (dmlc/treelite#617, dmlc/treelite#620)
* Support serializing models larger than `2**31 - 1` (dmlc/treelite#624)

Closes #7370
Closes #7368 
Closes #3838

In addition, update FIL to support vector bias from Treelite models.

Authors:
  - Philip Hyunsu Cho (https://github.com/hcho3)

Approvers:
  - Jim Crist-Harif (https://github.com/jcrist)
  - Robert Maynard (https://github.com/robertmaynard)
  - Simon Adorf (https://github.com/csadorf)

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

Labels

None yet

Projects

None yet

1 participant