https://github.com/radinshahdaei/text-classifier

Bert-based text classifier for movie genres.
https://github.com/radinshahdaei/text-classifier

bert pandas python sklearn

Last synced: about 5 hours ago
JSON representation

Bert-based text classifier for movie genres.

• Host: GitHub
• URL: https://github.com/radinshahdaei/text-classifier
• Owner: radinshahdaei
• Created: 2024-08-21T13:28:32.000Z (about 1 month ago)
• Default Branch: main
• Last Pushed: 2024-08-21T13:38:51.000Z (about 1 month ago)
• Last Synced: 2024-09-23T06:32:27.105Z (4 days ago)
• Topics: bert, pandas, python, sklearn
• Language: Jupyter Notebook
• Homepage:
• Size: 1.97 MB
• Stars: 0
• Watchers: 1
• Forks: 0
• Open Issues: 0
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

README

        
# Text Classifier

In this repository, you'll find a Jupyter Notebook which includes the code for a BERT-based text classifier. This model classifies the genre of a movie based on the summary of it in english or persian. The accuracy of this model isn't very high because of the biased and small dataset used for the project. An in-depth explaratory data analysis was done on the dataset prior to training the transformer models. some of the results on the accuracy of the model are shown below:

1) English text:



  



3) Persian text: