jbjeong91 / simple-text-summarizer

트랜스포머 기반 문서요약 모델, Transformer based text summarizer

Text-summarizer

Notes

• 트랜스포머를 기반으로하는 문장 요약 모델

  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.

Requirements

• Python3
• Tensorflow==1.12

Data

• 아마존 음식 리뷰 데이터

  Amazon review data is utilized

  Link: https://www.kaggle.com/snap/amazon-fine-food-reviews

How to use

• 데이터를 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
  

Example