search

BiomedSciAI / fuse-med-ml

A python framework accelerating ML based discovery in the medical field by encouraging code reuse. Batteries included :)
Apache License 2.0
134 stars 34 forks source link

Ehr transformer ICU example #245

Closed ellabarkan closed 1 year ago

ellabarkan commented 1 year ago

.

mosheraboh commented 1 year ago

This example is based on Vadim (@floccinauc) research. @ellabarkan implemented the dataset and tokenizer, and I implemented the model and training script. @michalozeryflato @SagiPolaczek , I'm merging now cause we need it for Sunday. But please review and add comments/questions to anyone of us.

mosheraboh commented 1 year ago

@ellabarkan , we still need to create a unittest. Can you please work on it in a new PR next week?