RNN 간이 영화 리뷰 분류


RNN 간이 영화 리뷰 분류

LSTM 과 CNN을 조합하여 'imdb' 에서 가져온 영화평을 토대로 긍정 or 부정의 두가지 결과만을 분류합니다. 사전에 데이터 전처리가 끝난 정보이기 때문에 원-핫 인코딩 과정과 상세 결과 출력 없이 정확도만을 보여주는 연습용 코드입니다. CNN 공부 다음으로 RNN 공부를 하고 있는데 정의, 원리부터 상세 이용까지 이전보다는 많이 어렵다는 느낌을 많이 받고 있습니다. 모듈 임포트 후 imdb에서 데이터를 가져옵니다. RNN과정에 쓰일 Embedding 과 LSTM도 볼 수 있습니다. 데이터 입력과 학습셋, 테스트셋 설정을 해줍니다. 주석 내용과 같이 첫 층에서 Embedding을 거친후, 과적합을 피하기 위해 드랍아웃을 실행, 이후 컨볼루션과 LSTM을 혼용하여 모델을 설정해줍니다. 출력층 활성화 함수는 시그모이드를 선택하였습니다. 학습을 진행합니다. 최종 출력 퍼센트는 두가지 방식으로 출력해봤습니다. 아무래도 포멧팅에 있어 밑에줄이 좀더 간결해 보입니다. 시그모이드 함수를 ...


#AI #텐서플로우 #인공지능 #연습 #머신러닝 #딥러닝 #RNN #LSTM #CNN #파이썬

원문링크 : RNN 간이 영화 리뷰 분류