tensorflow 输出层维度修改,关于return_sequences

def build_model():
    NUM_MIDDLE_01 = 100
    NUM_MIDDLE_02 = 120

    model = Sequential()
    model.add(LSTM(NUM_MIDDLE_01, input_shape=(train_x.shape[1], train_x.shape[2]) ,return_sequences=True))

    # model.add(LSTM(NUM_MIDDLE_01, input_shape=(NUM_LSTM, 1), return_sequences=True))
    model.add(Dropout(0.2))
    model.add(LSTM(NUM_MIDDLE_02))
    model.add(Dropout(0.2))
    model.add(Dense(1))
    model.add(Activation("linear"))
    _optimizer = tf.keras.optimizers.Adam(
        amsgrad=True
    )
    model.compile(loss='mae', optimizer=_optimizer)
    # model.compile(loss="mean_squared_error", optimizer=_optimizer)
    model.summary()
    # model.compile(loss="mse", optimizer='rmsprop')

    return model

Model: "sequential"
_________________________________________________________________
Layer (type)                 Output Shape              Param #   
=================================================================
lstm (LSTM)                  (None, 1, 100)            43600     
_________________________________________________________________
dropout (Dropout)            (None, 1, 100)            0         
_________________________________________________________________
lstm_1 (LSTM)                (None, 120)               106080    
_________________________________________________________________
dropout_1 (Dropout)          (None, 120)               0         
_________________________________________________________________
dense (Dense)                (None, 1)                 121       
_________________________________________________________________
activation (Activation)      (None, 1)                 0         
=================================================================
Total params: 149,801
Trainable params: 149,801
Non-trainable params: 0

return_sequences 为true,会产生多一维

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值