Skip to content

Add decision_function to return raw logits#23

Open
singjc wants to merge 1 commit intomesalock-linux:masterfrom
singjc:add/decision_function
Open

Add decision_function to return raw logits#23
singjc wants to merge 1 commit intomesalock-linux:masterfrom
singjc:add/decision_function

Conversation

@singjc
Copy link
Copy Markdown

@singjc singjc commented Nov 20, 2025

This PR adds a decision_function (same naming convention as scikit-learn), providing users with direct access to raw decision scores from the gradient boosting model.

The `decision_function` method computes the unnormalized raw scores (logits) from the gradient boosting model for the given test data. This allows the user to interpret the raw values directly, with higher values indicating stronger confidence in the positive class.

Example usage:
```rust
let raw_scores = gbdt.decision_function(&test_data);
```
@singjc
Copy link
Copy Markdown
Author

singjc commented Dec 16, 2025

Hi @dingelish , any updates on this being merged?

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