SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Add qas_id to SquadResult and SquadExample (#3745)

Browse Source 
* Add qas_id

* Fix incorrect name in squad.py

* Make output files optional for squad eval
This commit is contained in:
Jared T Nielsen
2020-04-20 14:08:57 -06:00
committed by GitHub
parent c4158a6314
commit c79b550dd0
3 changed files with 61 additions and 40 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
examples/run_squad.py
View File

@@ -307,7 +307,7 @@ def evaluate(args, model, tokenizer, prefix=""):
if args.model_type in ["xlm", "roberta", "distilbert", "camembert"]:
del inputs["token_type_ids"]
example_indices = batch[3]
feature_indices = batch[3]
# XLNet and XLM use more arguments for their predictions
if args.model_type in ["xlnet", "xlm"]:
@@ -320,8 +320,9 @@ def evaluate(args, model, tokenizer, prefix=""):
outputs = model(**inputs)
for i, example_index in enumerate(example_indices):
eval_feature = features[example_index.item()]
for i, feature_index in enumerate(feature_indices):
# TODO: i and feature_index are the same number! Simplify by removing enumerate?
eval_feature = features[feature_index.item()]
unique_id = int(eval_feature.unique_id)
output = [to_list(output[i]) for output in outputs]