bert-sentiment

KO | EN

gitlite — search

#python #java #python3 #arduino #golang #machine-learning #rust #html #flask #javascript #seismology #nodejs

★ 18 Open GitHub ↗

Sentiment analysis by BERT

// repository documentation

Was this content helpful?

★ 0 (0 ratings)

# Sentiment analysis by BERT in PyTorch
BERT is state-of-the-art natural language processing model from Google. Using its latent space, it can be repurpossed for various NLP tasks, such as sentiment analysis.

This simple wrapper based on [Transformers](https://github.com/huggingface/transformers) (for managing BERT model) and PyTorch achieves 92% accuracy on guessing positivity / negativity on IMDB reviews.

# How to use

## Prepare data

First, you need to prepare IMDB data which are publicly available. Format used here is one review per line, with first 12500 lines being positive, followed by 12500 negative lines. Or you can simply download dataset on my Google Drive [here](https://drive.google.com/drive/folders/1FiRODwhfJt6MpCqdfM7GgHwHqQ9VXFSJ?usp=sharing). Default folder read by script is `data/`.

## Train weights

Training with default parameters can be performed simply by.

`python script.py --train`

Optionally, you can change output dir for weights or input dir for dataset.

## Evaluate weights

You can find out how great you are (until your grandma gets her hands on BERT as well) simply by running

`python script.py --evaluate`

Of course, you need to train your data first or get them from my drive.

## Predict text

`python script.py --predict "It was truly amazing experience."`

`python script.py --predict "It was so terrible and disgusting as coffee topped with ketchup."`

bert-sentiment

Explore Similar Repositories

Was this content helpful?

Recent Feedback

Download README