Oufattole / meds-torch

MIT License
11 stars 1 forks source link

Support Multi-task learning #21

Open Oufattole opened 1 month ago

Oufattole commented 1 month ago

We currently support single task supervised training, we should support multiple tasks. Users would input a list of task parquet files (with patient_id, end-point timestamp, and labels) and we should join these dataframes and then integrate them into the pytorch_dataset class here.