Skip to content

Latest commit

 

History

3 Commits

Folders and files

NameName
Last commit message
Last commit date
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Using-bert-and-pytorch-for-IMDB

You can find the dataset on the kaggle :- https://www.kaggle.com/lakshmi25npathi/imdb-dataset-of-50k-movie-reviews

Train the file and then model will automatically saved with the best accuraccy. this model gives me accuracy upto 93%

And then for checking the results of the accuracy you can run the app.py file which will lead you to the flask server there you can check the positive reviews and negative reviews of the statements of the movies.

you can download the Bert_base_uncase/cased on hugging face itself.

About

No description, website, or topics provided.

Resources

Stars

3 stars

Watchers

1 watching

Forks

Releases

Packages

Contributors

Languages