【发布时间】:2018-08-16 23:23:29
【问题描述】:
我正在使用 Keras 在数据集上训练 LSTM 模型,如下所示。变量“描述”是一个文本字段,“年龄”和“性别”是分类和连续字段。
Age, Gender, Description
22, M, "purchased a phone"
35, F, "shopping for kids"
我正在使用词嵌入将文本字段转换为词向量,然后将其输入到 keras 模型中。代码如下:
model = Sequential()
model.add(Embedding(word_index, 300, weights=[embedding_matrix], input_length=70, trainable=False))
model.add(LSTM(300, dropout=0.3, recurrent_dropout=0.3))
model.add(Dropout(0.6))
model.add(Dense(1))
model.add(Activation('sigmoid'))
model.compile(loss='binary_crossentropy', optimizer='adam', metrics['accuracy'])
此模型运行成功,但我也想输入“年龄”和“性别”变量作为特征。代码中还需要哪些更改才能使用这些功能?
【问题讨论】:
-
您能告诉我您是如何输入包含
Age和Gender的文件的吗?
标签: python tensorflow machine-learning keras lstm