This commit is contained in:
Alan-Lee123
2017-06-16 09:33:38 +08:00
parent 57aa4bcf8e
commit 300bf19b59
2 changed files with 49 additions and 16 deletions
+5 -1
View File
@@ -22,9 +22,13 @@ def load_data(path):
return data
def tokenize_data(sdata, mxlen):
def init_tokenizer(sdata):
texts = [t[1] for t in sdata]
tokenizer.fit_on_texts(texts)
def tokenize_data(sdata, mxlen):
texts = [t[1] for t in sdata]
seqs = tokenizer.texts_to_sequences(texts)
seqs = pad_sequences(seqs, mxlen)
data = {}