rust-ml / rust-ml/linfa

get the score from logistic regression prediction

Open
#340 0 comments 0 reactions 0 assignees View on GitHub

Nobody has claimed this yet.

Dominant language
Rust
Stars
4.7k
Forks
334
Avg merge
39m
Merged PRs (30d)
1

Description

Hi,
We could get the prediction of logisticregression model using

let validation = model.set_threshold(threshold).predict(&dataset_validation);

However, can we get the prediction score of each records ? Then we can can put the prediction result into Excel and set the threshold mannually.

Contributor guide

No contributing guide indexed for this repository

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Research direction

Start at the logistic-regression model API around set_threshold(...).predict(&dataset_validation) and trace how predictions are produced. Determine the intended per-record score representation and expose it without thresholding; done means callers can export scores and choose a threshold externally.

Written by the indexing model from the issue text.

Assessment

Tech stack
rust
Domain
machine-learning
Issue type
Feature
Difficulty
4/5
Estimated time
3-5 days
Activity status
Stale
Clarity
Mostly clear
Newbie friendliness
38/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.