Skip to content

GitLab

Commit 43c24325 authored by erenup's avatar erenup
Browse files

avoid invalid labels of truth

parent 3c7e676f
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
examples/single_model_scripts/utils_multiple_choice.py
View file @ 43c24325
...@@ -249,6 +249,7 @@ class ArcProcessor(DataProcessor): ...@@ -249,6 +249,7 @@ class ArcProcessor(DataProcessor):
return int(truth) - 1 return int(truth) - 1
else: else:
logger.info("truth ERROR!") logger.info("truth ERROR!")
examples = [] examples = []
three_choice = 0 three_choice = 0
four_choice = 0 four_choice = 0
...@@ -267,6 +268,7 @@ class ArcProcessor(DataProcessor): ...@@ -267,6 +268,7 @@ class ArcProcessor(DataProcessor):
continue continue
four_choice += 1 four_choice += 1
truth = str(normalize(data_raw["answerKey"])) truth = str(normalize(data_raw["answerKey"]))
assert truth is not None
question_choices = data_raw["question"] question_choices = data_raw["question"]
question = question_choices["stem"] question = question_choices["stem"]
id = data_raw["id"] id = data_raw["id"]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment