LSTM Classifier

This project is a simple LSTM classifier that classifies news into 4 categories: world, business, sci/tech, and sports. The model was trained on the AG News dataset using a simple LSTM network, and using PyTorch.

Training code is available on Github

Checkout a little demo below deployed using AWS Lambda and API Gateway.

Classify news into 4 categories: world, business, sci/tech, and sports

Checkout the project in detail on Github