Skip to content

Conversation

@sreya-rc
Copy link

@sreya-rc sreya-rc commented Jan 2, 2026

Fixes #26596

  • Updated FinalizeScores1 methods in TreeAggregator, TreeAggregatorSum, and TreeAggregatorAverage to handle LOGISTIC transform
  • Added test case TreeEnsembleSingleTargetLogistic to verify correct behavior

Description

  • The LOGISTIC post-transform was not being applied to single-target TreeEnsemble regression models. The FinalizeScores1 methods only checked for PROBIT transform and ignored LOGISTIC, causing models to output raw scores instead of applying the logistic function.

Motivation and Context

  • According to the ONNX ML spec, TreeEnsemble operators should support LOGISTIC post-transform. However, for single-target regression models, only PROBIT was being applied. This caused models with post_transform=LOGISTIC to return raw prediction scores (10.0) instead of transformed values (0.9999546).

Fixes microsoft#26596

- Updated FinalizeScores1 methods in TreeAggregator, TreeAggregatorSum,
  and TreeAggregatorAverage to handle LOGISTIC transform
- Added test case TreeEnsembleSingleTargetLogistic to verify correct behavior
@sreya-rc
Copy link
Author

sreya-rc commented Jan 2, 2026

@microsoft-github-policy-service agree

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.

TreeEnsemble post_transform LOGISTIC does not work

1 participant