Optimizer comparison on stratified cross-validation data

greenelab / pancancer-evaluation

Evaluating genome-wide prediction of driver mutations using pan-cancer data

BSD 3-Clause "New" or "Revised" License

9 stars 3 forks source link

Optimizer comparison on stratified cross-validation data #79

Closed jjc2718 closed 1 year ago

jjc2718 commented 1 year ago

In order to separate the conclusion of the optimizer comparison into a separate small paper, we wanted to try it using stratified cross-validation (instead of holding out an entire cancer type or dataset); these changes enable that.

The results are mostly the same as we saw in #71: e.g. for KRAS:

And overall, liblinear performance is better than SGD for the best-performing model:

review-notebook-app[bot] commented 1 year ago

Check out this pull request on

See visual diffs & provide feedback on Jupyter Notebooks.