Update README.md

README.md

@@ -8,6 +8,31 @@ license: mit
 ## Description
 This model is a finetuned RoBERTa-based model pre-trained from scratch on Dutch hospital notes sourced from Electronic Health Records. All code used for the creation of MedRoBERTa.nl can be found at https://github.com/cltl-students/verkijk_stella_rma_thesis_dutch_medical_language_model. The publication associated with the negation detection task can be found at https://arxiv.org/abs/2209.00470. The code for finetuning the model can be found at https://github.com/umcu/negation-detection.
 
+
+## Minimal example
+
+```python
+tokenizer = AutoTokenizer.from_pretrained("UMCU/MedRoBERTa.nl_NegationDetection")
+model = AutoModelForTokenClassification.from_pretrained("UMCU/MedRoBERTa.nl_NegationDetection")
+
+some_text = "De patient was niet aanspreekbaar en hij zag er grauw uit. Hij heeft de inspanningstest echter goed doorstaan." 
+inputs = tokenizer(some_text, return_tensors='pt')
+output = model.forward(inputs)
+probas = torch.nn.functional.softmax(output.logits[0]).detach().numpy()
+
+#  koppel aan tokens
+input_tokens = tokenizer.convert_ids_to_tokens(inputs['input_ids'][0])
+target_map = {0: 'B-Negated', 1:'B-NotNegated',2:'I-Negated',3:'I-NotNegated'}
+results = [{'token': input_tokens[idx],
+                 'proba_negated': proba_arr[0]+proba_arr[2],
+                 'proba_not_negated': proba_arr[1]+proba_arr[3]
+                 }  
+                 for idx,proba_arr in enumerate(probas)]
+
+```
+
+It perhaps good to note that we assume the [Inside-Outside-Beginning](https://en.wikipedia.org/wiki/Inside%E2%80%93outside%E2%80%93beginning_(tagging)) format.
+
 ## Intended use
 The model is finetuned for negation detection on Dutch clinical text. Since it is a domain-specific model trained on medical data, it is meant to be used on medical NLP tasks for Dutch. This particular model is trained on a 512-max token windows surrounding the concept-to-be negated.