【发布时间】:2016-07-11 08:23:16
【问题描述】:
我现在正在尝试为基于 LSTM 的 NN 准备输入数据。 我有大量的文本文档,我想要为每个文档制作序列向量,这样我就可以将它们作为训练数据提供给 LSTM RNN。
我的糟糕做法:
import re
import numpy as np
#raw data
train_docs = ['this is text number one', 'another text that i have']
#put all docs together
train_data = ''
for val in train_docs:
train_data += ' ' + val
tokens = np.unique(re.findall('[a-zа-я0-9]+', train_data.lower()))
voc = {v: k for k, v in dict(enumerate(tokens)).items()}
然后 brutforce 用“voc”字典替换每个文档。
是否有任何库可以帮助完成这项任务?
【问题讨论】: