OHDSI Home | Forums | Wiki | Github

Examples of training deep learning models in python on OHDSI data?

(Jaan Altosaar) #1

It has been difficult to find examples that are written in python for using pytorch (or other deep learning frameworks) to train models on OHDSI data.

I was pointed to an R package that interfaces with pytorch: https://ohdsi.github.io/PatientLevelPrediction/articles/BuildingDeepLearningModels.html

Are there examples that are purely written in python? A colaboratory notebook would be ideal.

Thank you!

(David Sontag) #2

Here is our fully Python deep (and shallow) learning codebase for OMOP:


The deep model we implement is our AAAI '21 paper https://arxiv.org/abs/2007.05611, but the codebase will make it trivial for you to plug in others.

This is still in beta, so please do keep my lab in the loop as you test it out.

(Jaan Altosaar) #3

Fantastic, thanks so much David! Looking forward to reading the paper.