Briiick / imdb-classification

Take a pre-trained recent NLP model and apply it to classify the sentiment of movie reviews.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

IMDB BERT Sentiment Classification

Please see the notebook IMDB BERT Sentiment Classification.ipynb for details.

In this project, I seek to use DistilBERT and various machine learning techniques to classify good and bad reviews from the Kaggle IMDB dataset.

About

Take a pre-trained recent NLP model and apply it to classify the sentiment of movie reviews.


Languages

Language:Jupyter Notebook 100.0%