-
Notifications
You must be signed in to change notification settings - Fork 67
Seems really difficult to reproduce the results #6
Comments
Hi mahi83, Thanks for trying out the code.
Best, |
I used the same hyper parameters as mentioned in the appendix of the paper. I was running into errors when running other folds. I didn't spend time trying to debug the errors, as I was focused on the first fold. |
Hi @mp2893 I'm also receiving errors running on Fold >= 1 using the paper's hyperparameters for the readmission prediction task. In Fold 0, my AUC-PR (~0.3) and AUC-ROC (~0.6) are both seem lower than in the published results. Running Fold 0 again now and here are metrics around ~160k steps. |
Hi @foadgr, I completely forgot about this issue. |
Sounds great, will give this a try. Thank you! |
I have been trying to replicate the default code you all set, where we want to predict readmission on "5-folds". I have run this scrip a couple of times on 1 million steps on "Fold 0". For some reason, I can't get any AUC-PR score near the reported score. I have been getting an AUC-PR of around 0.3. However, the paper reports an AUC score of around 0.5244. Any comments of changes I need to make to get similar results?
The text was updated successfully, but these errors were encountered: