Browse Source

fix a bug in label2index dict.

tags/v0.1.0
2017alan 6 years ago
parent
commit
c24d01d50f
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      fastNLP/core/preprocess.py

+ 2
- 2
fastNLP/core/preprocess.py View File

@@ -239,7 +239,7 @@ class SeqLabelPreprocess(BasePreprocess):
label2index: dict of {str, int}
"""
# In seq labeling, both word seq and label seq need to be padded to the same length in a mini-batch.
label2index = DEFAULT_WORD_TO_INDEX.copy()
label2index = {} # DEFAULT_WORD_TO_INDEX.copy()
word2index = DEFAULT_WORD_TO_INDEX.copy()
for example in data:
for word, label in zip(example[0], example[1]):
@@ -297,7 +297,7 @@ class ClassPreprocess(BasePreprocess):

# build vocabulary from scratch if nothing exists
word2index = DEFAULT_WORD_TO_INDEX.copy()
label2index = DEFAULT_WORD_TO_INDEX.copy()
label2index = {} # DEFAULT_WORD_TO_INDEX.copy()

# collect every word and label
for sent, label in data:


Loading…
Cancel
Save