Commit ce1389c9 authored by lintangsutawika's avatar lintangsutawika
Browse files

doc_to_target set to just label

parent dd8bbe2c
...@@ -6,7 +6,7 @@ output_type: multiple_choice ...@@ -6,7 +6,7 @@ output_type: multiple_choice
validation_split: validation validation_split: validation
test_split: test test_split: test
doc_to_text: !function utils.doc_to_text_et doc_to_text: !function utils.doc_to_text_et
doc_to_target: !function utils.doc_to_target doc_to_target: label
doc_to_choice: !function utils.doc_to_choice doc_to_choice: !function utils.doc_to_choice
metric_list: metric_list:
- metric: acc - metric: acc
...@@ -11,12 +11,6 @@ def doc_to_text(doc, connector): ...@@ -11,12 +11,6 @@ def doc_to_text(doc, connector):
return doc["premise"].strip()[:-1] + f" {conn}" return doc["premise"].strip()[:-1] + f" {conn}"
def doc_to_target(doc):
correct_choice = doc["choice1"] if doc["label"] == 0 else doc["choice2"]
# Connect the sentences
return convert_choice(correct_choice)
def doc_to_choice(doc): def doc_to_choice(doc):
return [convert_choice(doc["choice1"]), convert_choice(doc["choice2"])] return [convert_choice(doc["choice1"]), convert_choice(doc["choice2"])]
......
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