Skip to content

Practicing implementation of text sentiment analysis models and huggingface packages such as Transformers and Datasets, using the IMDB review dataset.

Notifications You must be signed in to change notification settings

slyyyle/distilbert-finetuning-imdb

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

6 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

distilbert-finetuning-imdb

This was a relatively "on-the-rails" example which allowed me to re-acquaint with the huggingface package ecosystem.

Contents:

imdb_sentiment.ipynb - Jupyter notebook used to finetune distilbert-base to detect binary sentiment (good/bad) from IMDB reviews. This problem was quite simple and thus, the results were quite good after finetuning.

IMDB Dataset.csv - CSV dataset containing IMDB reviews, labeled for sentiment analysis (good/bad).

About

Practicing implementation of text sentiment analysis models and huggingface packages such as Transformers and Datasets, using the IMDB review dataset.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published