zli12321 commited on
Commit
08f9b94
1 Parent(s): d5b56f2

Update README.md

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