search

som-shahlab / ehr_ml

Code for doing machine learning with various EHRs
MIT License
21 stars 3 forks source link

Jlli/refactor #17

Closed woffett closed 3 years ago

woffett commented 3 years ago

Refactor model pre-training and featurization into a sciki-like API

woffett commented 3 years ago

I'm still working on the interface for the featurizer.fit method so that people don't have to instantiate a StrideDataset when they want to pretrain. Ideally they can pass in the same things for fit and featurize.