Thanks to visit codestin.com
Credit goes to github.com

Skip to content

Add wrapper class that changes threshold value for predict #8614

Closed
@amueller

Description

@amueller

This was discussed before, but not sure if there's an issue.
We should have a wrapper class that changes the decision threshold based on a cross-validation (or hold-out, pre-fit) estimate.
This is very very common so I think we should have a built-in solution.
Simple rules for selecting a new threshold are:

  • picking the point on the roc curve that's closest to the ideal corner
  • picking the point on the precision-recall curve that's closest to the ideal corner
  • optimize one metric while holding another one constant: find the best threshold to yield the best recall with a precision of at least 10% for example. We could also make this slightly less flexible and just say "the median of the largest threshold over cross-validation that yields at least X precision".

Metadata

Metadata

Assignees

No one assigned

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions