uf-hobi-informatics-lab / ClinicalTransformerRelationExtraction

A project for developing transformer-based models for clinical relation extraction
MIT License
125 stars 23 forks source link

Not able to get the prediction for Test.csv #23

Closed vikasgoel2000 closed 2 years ago

vikasgoel2000 commented 2 years ago

Hi

I am just trying to run the code to get the predictions for the test.csv. i am trying with the pre trained model at https://transformer-models.s3.amazonaws.com/mimiciii_bert_10e_128b.zip.

While running code I am getting an error as AttributeError: 'BertConfig' object has no attribute 'tags'

Screen shot of my scree is as below

image

bugface commented 2 years ago

the pretrained model is pretrained on MLM, not on RE so it is not an out-of-box RE model for inference.

You need to first train a RE model on training data during which the tags and other configs will be generated based on the dataset and parameters, then you can run inference.