Inferencing for custom dataset
Siddhijain16 opened this issue · 2 comments
Siddhijain16 commented
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 !!
BruceWen120 commented
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.
BruceWen120 commented
I will close this now since there's no activity. Feel free to reopen if you have new questions.