개발

LSTM stateful

murbachovski 2024. 3. 21. 22:03

https://tykimos.github.io/2017/04/09/RNN_Layer_Talk/

 

순환 신경망 모델 만들어보기

앞서 살펴본 LSTM 레이어를 이용하여 몇가지 순환 신경망 모델을 만들어보고, 각 모델에 “나비야” 동요를 학습시켜보면서 자세히 살펴보겠습니다. 시퀀스 데이터 준비 순환 신경망은 주로 자

tykimos.github.io

오랜만에 현장 강의를 들으니 유익하고 즐거웠다.

비전 강의를 들으면서 LSTM모델을 설계할때 stateful의 존재를 알게 됐다.

model = Sequential()
model.add(LSTM(128, batch_input_shape = (1, 4, 1), stateful=True))
model.add(Dense(one_hot_vec_size, activation='softmax'))

default => stateful=False

 

추후에 TimeStep을 길게 가져가야하는 모델을 개발할때 유용한 정보이다.