greenelab / netscape

Exploring the network/pathway regularization landscape for gene expression data
BSD 3-Clause "New" or "Revised" License
0 stars 1 forks source link

PyTorch logistic regression - implementation and comparison against scikit-learn #11

Closed jjc2718 closed 5 years ago

jjc2718 commented 5 years ago

This PR includes:

This implementation will lay the groundwork for future PRs, which will extend these analyses to more complex models (multi-layer NNs, and/or custom loss functions incorporating pathway and network priors).

Results indicate that prediction quality (as measured by accuracy/AUROC/AUPRC) is approximately the same for both models, but the scikit-learn models are much sparser (many more 0 coefficients). Planning to look into this in future PRs, in addition to running the same experiments for other genes besides TP53.

image