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

Fix tf text class (#7724)

Browse Source 
* Fix test

* fix generic text classification

* fix test

* Fix tests
This commit is contained in:
Julien Plu
2020-10-12 14:45:15 +02:00
committed by GitHub
parent d6175a4268
commit d9ffb87efb
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/text-classification/run_tf_text_classification.py
View File

@@ -60,7 +60,7 @@ def get_tfds(
for k in files.keys(): for k in files.keys():
transformed_ds[k] = ds[k].map( transformed_ds[k] = ds[k].map(
lambda example: tokenizer.batch_encode_plus( lambda example: tokenizer.batch_encode_plus(
(example[features_name[0]], features_name[1]), (example[features_name[0]], example[features_name[1]]),
truncation=True, truncation=True,
max_length=max_seq_length, max_length=max_seq_length,
padding="max_length", padding="max_length",