LSTM使用生成器函数

时间:2018-05-24 21:53:16

标签: python neural-network keras lstm recurrent-neural-network

我正在尝试创建一个模型,其LSTM层为100个单位,输入尺寸为(16,48,12)(16是通过生成器函数输入的批量大小)。生成器函数产生预期的输出(16,1,2)(16是批量大小),我想用作具有softmax激活功能的密集层的输出。最好的方法是什么?我对keras很新,我无法安静地掌握使用生成器函数......

1 个答案:

答案 0 :(得分:0)

就个人而言,建议在完全连接的密集层之前使用PReLU激活功能。 例如: model.add(LSTM(128,input_shape=(train_X.shape[1],train_X.shape[2]))) model.add(BatchNormalization()) model.add(Dropout(.2)) model.add(Dense(64)) model.add(PReLU())