McGill-NLP/medal

Inferencing for custom dataset

Siddhijain16 opened this issue · 2 comments

Hi @BruceWen120 ,
How do I use this model for my own healthcare dataset?
Please share the steps how to do the inferencing from this model.
Thank you !!

Thank you for your interest.

Depending on your dataset and task, the code in the downstream folder may or may not be directly applicable. If your task requires a prediction per input sentence, this is similar to the MIMIC tasks we've done, and you can easily adapt the code by providing the correct input and labels.

Otherwise, you can take the final hidden representations from the pre-trained model and build your own output layer on them.

I will close this now since there's no activity. Feel free to reopen if you have new questions.