vanderschaarlab / autoprognosis

A system for automating the design of predictive modeling pipelines tailored for clinical prognosis.
https://www.autoprognosis.vanderschaar-lab.com/
Apache License 2.0
119 stars 27 forks source link

Possibility to perform upsampling #75

Closed jbwoillard closed 2 months ago

jbwoillard commented 1 year ago

Dear mainteners, we tested autoprognosis with one of our dataset with class highly imbalanced (event proportion =16%). Autoprognosis returned values of sensitivity to 0 and specificity to 1(classification of all patients as non event). Would it be possible to add an upsampling step in the AutoML solution for these specific case or to change the optimisation metrics? Thanks in advance All the best jb

DrShushen commented 2 months ago

This is not currently planned unfortunately but you can use other open source libraries like sklearn, imblearn to perform this on your dataset before using autoprognosis