bert-uncased-news-classifier

Fine-tuned google-bert-uncased model

The model is fine-tuned using 210k huffington post articles into 42 categories.

The model can handle a maximum token length of 256

The model was fine-tuned using 60 / 20 / 20 stratified splits for training, eval and test sets

The model achieves a performance of: </br> eval_loss: 1.719 </br> eval_f1: 0.776 </br> epoch: 10.0 </br>

The model performance based on weighted F1-score across 42 categories is: </br> Weighted F1 Score on Test Set: 0.7009

Examples Predictions: </br> Text: Manage Your Expectations, Lower Your Stress Your outlook is a product of your own relationship with expectation. What will or won't happen, no one knows. And how we deal with the stress of not knowing, whether to hope for the best or expect the worst, the idea that our expectations always directly affect an outcome is little more than magical thinking. </br> Predicted label: WELLNESS, Score: 0.7709

Text: Donald Trump Tells Drought-Plagued Californians: 'There Is No Drought' "If I win, believe me, we’re going to start opening up the water." </br> Predicted label: POLITICS, Score: 0.9552

Text: Sticker Shock: 5 Of The World's Most Expensive Hotel Suites (PHOTOS) You can only imagine our heart palpitations when we see the nightly rates of these hotels' top-of-the-line suites. </br> Predicted label: TRAVEL, Score: 0.9818