代码之家  ›  专栏  ›  技术社区  ›  Stupid420

如何添加注意层二进制分类RNN

  •  1
  • Stupid420  · 技术社区  · 7 年前

    我想把注意力放在第一层之后。下面是代码。

    visible = Input(shape=(251,))    
    embed=Embedding(vocab_size, 50)(visible)
    bilstm = Bidirectional(LSTM(units=25, return_sequences=True))(embed)    
    att==??    
    predictions=Dense(1, activation='sigmoid')(att)
    

    0 回复  |  直到 7 年前