Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
chenpangpang
transformers
Commits
0513f8d2
"vscode:/vscode.git/clone" did not exist on "afb050b29d0cac27c32c19c8206a9ac2a4662de2"
Unverified
Commit
0513f8d2
authored
Aug 03, 2020
by
Victor SANH
Committed by
GitHub
Aug 03, 2020
Browse files
correct label extraction + add note on discrepancies on trained MNLI model and HANS (#6221)
parent
3c289fb3
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
6 additions
and
2 deletions
+6
-2
examples/adversarial/utils_hans.py
examples/adversarial/utils_hans.py
+6
-2
No files found.
examples/adversarial/utils_hans.py
View file @
0513f8d2
...
@@ -255,7 +255,11 @@ class HansProcessor(DataProcessor):
...
@@ -255,7 +255,11 @@ class HansProcessor(DataProcessor):
return
self
.
_create_examples
(
self
.
_read_tsv
(
os
.
path
.
join
(
data_dir
,
"heuristics_evaluation_set.txt"
)),
"dev"
)
return
self
.
_create_examples
(
self
.
_read_tsv
(
os
.
path
.
join
(
data_dir
,
"heuristics_evaluation_set.txt"
)),
"dev"
)
def
get_labels
(
self
):
def
get_labels
(
self
):
"""See base class."""
"""See base class.
Note that we follow the standard three labels for MNLI
(see :class:`~transformers.data.processors.utils.MnliProcessor`)
but the HANS evaluation groups `contradiction` and `neutral` into `non-entailment` (label 0) while
`entailment` is label 1."""
return
[
"contradiction"
,
"entailment"
,
"neutral"
]
return
[
"contradiction"
,
"entailment"
,
"neutral"
]
def
_create_examples
(
self
,
lines
,
set_type
):
def
_create_examples
(
self
,
lines
,
set_type
):
...
@@ -268,7 +272,7 @@ class HansProcessor(DataProcessor):
...
@@ -268,7 +272,7 @@ class HansProcessor(DataProcessor):
text_a
=
line
[
5
]
text_a
=
line
[
5
]
text_b
=
line
[
6
]
text_b
=
line
[
6
]
pairID
=
line
[
7
][
2
:]
if
line
[
7
].
startswith
(
"ex"
)
else
line
[
7
]
pairID
=
line
[
7
][
2
:]
if
line
[
7
].
startswith
(
"ex"
)
else
line
[
7
]
label
=
line
[
-
1
]
label
=
line
[
0
]
examples
.
append
(
InputExample
(
guid
=
guid
,
text_a
=
text_a
,
text_b
=
text_b
,
label
=
label
,
pairID
=
pairID
))
examples
.
append
(
InputExample
(
guid
=
guid
,
text_a
=
text_a
,
text_b
=
text_b
,
label
=
label
,
pairID
=
pairID
))
return
examples
return
examples
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment