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