havakv / havakv/pycox

Event weighting for imbalanced datasets

Open
#100 1 comment 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
995
Forks
203
PR merge metrics
No merged PRs in 30d

Description

Thanks for the great library! I'm training with an imbalanced dataset. In time-to-event prediction, is it reasonable to weigh positive examples, as can be done in the classification analog? If so, could a convenience keyword arg to the loss functions be added (see `pos_weight` [here](https://pytorch.org/docs/stable/generated/torch.nn.BCEWithLogitsLoss.html))? I think this would be similar to, but more efficient than, oversampling positive examples.

Contributor guide

No contributing guide indexed for this repository

Research direction

Start by reviewing the repository's existing loss functions and the PyTorch BCEWithLogitsLoss documentation for `pos_weight`. Determine how event weighting should apply across the survival losses and how it should compare with positive-example oversampling; the work is done when the supported keyword behavior is defined and consistently implemented.

Written by the indexing model from the issue text.

Assessment

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

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.