search

Bcromas / propaganda_bert

Predicting propaganda in news articles using BERT.
3 stars 0 forks source link

Address class imbalance with SMOTE #6

Open Bcromas opened 4 years ago

Bcromas commented 4 years ago

Classes are imbalanced. Currently addressed in logistic reg through class_weights param but not directly addressed in BERT. Could generate a balanced dataset & evaluate results on it.

Bcromas commented 4 years ago

from imblearn.over_sampling import SMOTE