(keras) Timedistributed Layer

시계열 데이터를 LSTM 혹은 BILSTM으로 학습을 하고

최종 Output Layer가 many to many 일경우 마지막 층으로 Dense나 기타다른 층으로 연결하고 싶었다.

 

하지만 LSTM(BILSTM)은 출력층이 시퀀스 데이터이므로 Dense Layer를 그대로 사용하지 못한다. 

그럴때 사용하는 레이어가 바로 Timedistributed Layer이다. 

(이 레이어가 특정 레이어를 시계열형식의 연속적인 형태로 만들어주는 기능을 하는것 같다.)

 

예시)

from tensorflow.keras.layers import *
 
bilstm = Bidirectional(LSTM(100,return_sequences=True))(emb)
tag = TimeDistributed(Dense(len(tag.keys()),activation='softmax'))(bilstm)