AyoubChLin commited on
Commit
391f89c
1 Parent(s): 5d80da4

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +5 -2
README.md CHANGED
@@ -16,9 +16,12 @@ widget:
16
 
17
  This repository contains a fine-tuned DistilBert 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.
18
 
19
- The model was fine-tuned for four epochs, achieving a training loss of 0.052900, a validation loss of 0.257164,
20
- and a validation accuracy of 0.960415.
21
 
 
 
 
 
22
 
23
  ### Model Description
24
 
 
16
 
17
  This repository contains a fine-tuned DistilBert 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.
18
 
19
+ The model was fine-tuned for four epochs achieving a training loss of 0.012900, a validation loss of 0.151663,
 
20
 
21
+ - accuracy of 0.9607394366197183.
22
+ - f1 : 0.962072
23
+ - precision : 0.961904
24
+ - recall : 0.962324
25
 
26
  ### Model Description
27