diff --git a/fastNLP/core/preprocess.py b/fastNLP/core/preprocess.py index 99bf45ba..1c419ce9 100644 --- a/fastNLP/core/preprocess.py +++ b/fastNLP/core/preprocess.py @@ -268,7 +268,7 @@ class ClassPreprocess(BasePreprocess): for word in sent: if word not in word2index: - word2index[word[0]] = len(word2index) + word2index[word] = len(word2index) return word2index, label2index def to_index(self, data):