美文网首页
LSTM 返回状态

LSTM 返回状态

作者: 杨连坤 | 来源:发表于2019-07-08 22:51 被阅读0次
    def gru_model(units1=[9,196,196*2,196*4,196]):
    
        inputs=Input(shape=(units1[0],units1[1]))
    
        lstm=GRU(units=units1[2],return_sequences=True)(inputs)
    
        lstm=Dropout(rate=0.5)(lstm)
    
        lstm=GRU(units=units1[3],return_sequences=True)(lstm)
    
        lstm=Dropout(rate=0.5)(lstm)
    
        lstm=GRU(units=units1[4],return_sequences=True,return_state=True)(lstm)
    
    #  return_sequences=true 返回整个序列 也就是输入n个时间步 设定uints=m 返回[none,n,m]
    
    #  return_sequences=false  只返回最后一个状态  设定uints=m 返回[none,m]
    
    #  return_state=True  返回最后一个状态 与 return_sequences=false 返回相同 [[],[]]
    
    #  return_state=True  只返回序列 无最后状态
    
    
    
        outputs=lstm
    

    相关文章

      网友评论

          本文标题:LSTM 返回状态

          本文链接:https://www.haomeiwen.com/subject/mhmzhctx.html