zli12321 commited on
Commit
5d860ed
1 Parent(s): eecb606

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -189,7 +189,7 @@ from qa_metrics.transformerMatcher import TransformerMatcher
189
 
190
  question = "Which movie is loosley based off the Brother Grimm's Iron Henry?"
191
  # Supported models: roberta-large, roberta, bert, distilbert, distilroberta
192
- tm = TransformerMatcher("roberta-large")
193
  scores = tm.get_scores(reference_answer, candidate_answer, question)
194
  match_result = tm.transformer_match(reference_answer, candidate_answer, question)
195
  print("Score: %s; bert Match: %s" % (scores, match_result))
 
189
 
190
  question = "Which movie is loosley based off the Brother Grimm's Iron Henry?"
191
  # Supported models: roberta-large, roberta, bert, distilbert, distilroberta
192
+ tm = TransformerMatcher("zli12321/answer_equivalence_tiny_bert")
193
  scores = tm.get_scores(reference_answer, candidate_answer, question)
194
  match_result = tm.transformer_match(reference_answer, candidate_answer, question)
195
  print("Score: %s; bert Match: %s" % (scores, match_result))