-
트랜스포머를 기반으로하는 문장 요약 모델
The code is for Text summery based on Transformer and Point Generative Network
-
이 코드는 아래의 Transforemr 코드를 수정하여 제작되었음
The codes are modified based on the Transformer repo. Some parts from original repo have been deleted and modified.
- Python3
- Tensorflow==1.12
-
아마존 음식 리뷰 데이터
Amazon review data is utilized
-
데이터를 Data 링크에서 다운로드하여 Data in 폴더에 csv 파일 넣기
Down load the Data from the link and put in the 'Data in' folder.
-
Train
python main.py --train --train_steps=30000 --batch_size=32
-
Test
python main.py --test