pere commited on
Commit
8c33689
·
1 Parent(s): 4abafd7
Files changed (2) hide show
  1. __pycache__/tasks.cpython-38.pyc +0 -0
  2. tasks.py +2 -2
__pycache__/tasks.cpython-38.pyc CHANGED
Binary files a/__pycache__/tasks.cpython-38.pyc and b/__pycache__/tasks.cpython-38.pyc differ
 
tasks.py CHANGED
@@ -12,7 +12,7 @@ import tensorflow.compat.v1 as tf
12
  tsv_path = {
13
  "train": "gs://north-t5x/corpus/danish_hate/train.tsv",
14
  "validation": "gs://north-t5x/corpus/danish_hate/eval.tsv",
15
- "test": "gs://north-t5x/corpus/danish_hate/test.tsv"
16
  }
17
 
18
  vocabulary = seqio.SentencePieceVocabulary(
@@ -57,7 +57,7 @@ seqio.TaskRegistry.add(
57
  preprocessors=[
58
  functools.partial(
59
  t5.data.preprocessors.parse_tsv,
60
- field_names=["source","target"]),
61
  categorise_preprocessor,
62
  seqio.preprocessors.tokenize_and_append_eos,
63
  ],
 
12
  tsv_path = {
13
  "train": "gs://north-t5x/corpus/danish_hate/train.tsv",
14
  "validation": "gs://north-t5x/corpus/danish_hate/eval.tsv",
15
+ "test": "gs://north-t5x/corpus/danish_hate/eval.tsv"
16
  }
17
 
18
  vocabulary = seqio.SentencePieceVocabulary(
 
57
  preprocessors=[
58
  functools.partial(
59
  t5.data.preprocessors.parse_tsv,
60
+ field_names=["id","source","target"]),
61
  categorise_preprocessor,
62
  seqio.preprocessors.tokenize_and_append_eos,
63
  ],