Skip to content

xgboost converter updates#106

Open
thesps wants to merge 4 commits into
masterfrom
xgb_3_converter_fix
Open

xgboost converter updates#106
thesps wants to merge 4 commits into
masterfrom
xgb_3_converter_fix

Conversation

@thesps

@thesps thesps commented May 11, 2026

Copy link
Copy Markdown
Owner

The XGBoost converter hasn't worked that well since XGBoost version 2 (now at 3.3.0). This PR implements some missed feature - the model initial prediction / base score, and updates the tests.

Before this update, using the new test script, all of the tests failed based on numerical mismatches. Now, 3 of the tests pass and 3 of the tests fail due to leaf mismatches. I observed that even a single tree model would show numerical mismatches - hinting that reaching a different leaf was an issue as well as the different numerical values from the missing base score. The issue of reaching a wrong leaf is not fixed by this PR, but the testing is now able to help isolate the two types of issues.

Reaching a wrong leaf will be addressed in a future update, and seems to occur specifically for values which are very close to a decision threshold.

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