AyoubChLin commited on
Commit
87c7fd8
1 Parent(s): e282d16

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +6 -1
README.md CHANGED
@@ -12,7 +12,12 @@ pipeline_tag: text-classification
12
 
13
  This repository contains a fine-tuned Bert base model for sequence classification on the CNN News dataset. The model is able to classify news articles into one of six categories: business, entertainment, health, news, politics, and sport.
14
 
15
- The model was fine-tuned for four epochs achieving a validation loss of 0.190814, accuracy of 0.956690, precision of 0.956393, recall of 0.956690, and F1-score of 0.956144.
 
 
 
 
 
16
 
17
  ## Model Description
18
 
 
12
 
13
  This repository contains a fine-tuned Bert base model for sequence classification on the CNN News dataset. The model is able to classify news articles into one of six categories: business, entertainment, health, news, politics, and sport.
14
 
15
+ The model was fine-tuned for four epochs achieving a training loss of 0.077900, a validation loss of 0.190814
16
+
17
+ - accuracy : 0.956690.
18
+ - f1 : 0.956144.
19
+ - precision : 0.956393
20
+ - recall : 0.956690
21
 
22
  ## Model Description
23