Sequence classification model for sentiment analysis of Russian conversational text

Project structure

notebooks

.ipynb files

• csv_preprocessing.ipynb - pre processing initial csv files
• eda.ipynb - exploratory data analysis
• training.ipynb - model training
• evaluation.ipynb - model evaluation

src

.py files

• training.py contains Trainer class to train BertForSequenceClassification
• prediction.py contains Predictor class to predict sentiments from text data using pre trained model
• metrics.py contains functions to compute precision, recall, and f1 scores
• evaluation.py contains a function to evaluate pre trained model using test data
• input_data_preprocessing.py contains utility function to fix initial csv files

Metrics