Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Request to enable sample weights for Cox and AFT objectives #2637

Open
rob7113 opened this issue Apr 16, 2024 · 1 comment
Open

Request to enable sample weights for Cox and AFT objectives #2637

rob7113 opened this issue Apr 16, 2024 · 1 comment

Comments

@rob7113
Copy link

rob7113 commented Apr 16, 2024

Problem: Cox and AFT objectives currently do not support sample weights, yet enabling them would considerably broaden the applicability of survival analysis with Catboost, for instance, by allowing censored and uncensored data to have different weights.
catboost version: 1.2.3
Operating System: linux
CPU: second generation AMD EPYC
GPU: NVIDIA

@rob7113
Copy link
Author

rob7113 commented Apr 16, 2024

I've been training several survival models with CatBoost recently, playing around with both the Cox and AFT objectives. The latter is especially suited for my application purposes and trains very smoothly. However, my analysis is limited because sample weights are not supported by any of these two objectives. Enabling sample weights would broaden the applicability of Catboost to survival analysis and the required code changes should be straightforward, since sample weights are already supported by other objectives.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

No branches or pull requests

2 participants