lighteternal
/

BiomedNLP-PubMedBERT-base-uncased-abstract-fulltext-finetuned-mnli

Text Classification

textual-entailment

Model card Files Files and versions

Metrics Training metrics Community

lighteternal commited on Oct 26, 2021

Commit

4e4ff12

·

1 Parent(s): 40b7109

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -38,7 +38,7 @@ hypothesis = 'EpCAM is downregulated in breast cancer.'
 # run through model pre-trained on MNLI
 x = tokenizer.encode(premise, hypothesis, return_tensors='pt',
                      truncation_strategy='only_first')
-logits = model(x.to(device))[0]
 probs = logits.softmax(dim=1)
 print('Probabilities for entailment, neutral, contradiction \n', np.around(probs.cpu().

 # run through model pre-trained on MNLI
 x = tokenizer.encode(premise, hypothesis, return_tensors='pt',
                      truncation_strategy='only_first')
+logits = model(x)[0]
 probs = logits.softmax(dim=1)
 print('Probabilities for entailment, neutral, contradiction \n', np.around(probs.cpu().