Skip to content

Sentiment Analysis of imdb movie comments, implemented by transformers (Using Pytorch)

Notifications You must be signed in to change notification settings

1989ONCE/BERT-Model

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 

Repository files navigation

BERT-Model

Sentiment Analysis implement by transformers (Using Pytorch)

Datasets:

  • load_dataset-> imdb movie comments
  • split into three parts: train, validation, test
  • label def:
    • 0 for negative comment
    • 1 for positive comment

BERT MODEL

-> bert-base-uncased: 93%

Best Result of Test Accuracy

Test Accuracy = 0.93

Prediction Result for test data

image

Random Test for random sentence

Result was Correct, "This movie doesn't attract me" IS A negative comment.
image

About

Sentiment Analysis of imdb movie comments, implemented by transformers (Using Pytorch)

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published