Skip to content
This repository has been archived by the owner on Sep 29, 2023. It is now read-only.

Streamline featurize and add a BaseModelOutput type class #10

Open
jason-fries opened this issue May 8, 2021 · 0 comments
Open

Streamline featurize and add a BaseModelOutput type class #10

jason-fries opened this issue May 8, 2021 · 0 comments
Assignees
@EthanSteinberg @woffett
Labels
enhancement New feature or request

Comments

@jason-fries
Copy link
Contributor

jason-fries commented May 8, 2021 •
edited
Loading

ehr_ml.clmbr.featurize_patients should be refactored to improve the interface and enrich the outputs of the model akin to
https://huggingface.co/transformers/main_classes/output.html#basemodeloutput

Simple pseudo-code

model = clmbr.from_pretrained(PATH_TO_CLMBR_MODEL)
train = dataloaders.Timelines(PATH_TO_EXTRACT, patient_cohort_dataframe) 
x_enc = model(train)
@jason-fries jason-fries added the enhancement New feature or request label May 8, 2021
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Assignees

@EthanSteinberg EthanSteinberg

@woffett woffett

Labels
enhancement New feature or request
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
3 participants
@EthanSteinberg @jason-fries @woffett