NainiShah / News-Headline-Generation

Our goal is to implement text summarization by generating headline for a news body using recurrent neural networks. We attempt to reproduce the results in the paper: https://arxiv.org/pdf/1512.01712.pdf

News Headline Generator

We have built a web application using flask framework in python that generates headline for the given news body. This application can be used by bloggers, journalists etc. where they can easily come up with the summarized headline for the article/blog they are working upon. We attempt to reproduce the results in the paper: Generating News Headlines with Recurrent Neural Networks

How to run

Software:

• Install python: Official website
• Install anaconda: Official website
• Install Keras
• Install Tensorflow
• Neural networks are computations heavy, GPU configuration is recommended.

Data

We have used All the news dataset from kaggle Extract only title and content from the dataset. Later, we tokenized the title,content and saved it as a pickle file. We divided our data set into train, validation and test sets respectively.

Train / test

Train() method inside “ ” notebook describes how the model is trained using Keras Test() method inside “ ” generates headline from the body inputted via test file