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
网友评论