output_token_type -> token_type_ids

This commit is contained in:
LysandreJik committed 2019-09-24 07:21:38 -04:00
1 parent 3927d7756c
commit c832f43a4d
3 files changed
+5 -5

No files matched your search

+1 -1
View File
@@ -413,7 +413,7 @@ def convert_examples_to_features(examples, label_list, max_seq_length,
max_length=max_seq_length,
truncate_first_sequence=True # We're truncating the first sequence as a priority
)
input_ids, segment_ids = inputs["input_ids"], inputs["output_token_type"]
input_ids, segment_ids = inputs["input_ids"], inputs["token_type_ids"]
# The mask has 1 for real tokens and 0 for padding tokens. Only real
# tokens are attended to.