AyoubChLin commited on
Commit
f94d9d0
1 Parent(s): 576d211

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +8 -5
README.md CHANGED
@@ -10,7 +10,10 @@ pipeline_tag: text-classification
10
  ---
11
 
12
  # Model Description
13
- This is a fine-tuned DistilBART model for sequence classification on CNN news articles for text classification. The model was fine-tuned using a batch size of 32, a learning rate of 6e-5, and for 1 epoch.
 
 
 
14
 
15
  ## Dataset
16
  The CNN News dataset was used for fine-tuning the model. The dataset consists of news articles from various categories such as sports, entertainment, politics, etc.
@@ -18,10 +21,10 @@ The CNN News dataset was used for fine-tuning the model. The dataset consists of
18
  ## Performance
19
  The following performance metrics were achieved after fine-tuning the model:
20
 
21
- Accuracy: 0.9597114707952147
22
- F1-score: 0.9589247895703302
23
- Recall: 0.9597114707952147
24
- Precision: 0.9589649408501851
25
 
26
  ## Usage
27
  You can use this model to classify CNN news articles into different categories such as sports, entertainment, politics, etc. You can load the model using the Hugging Face Transformers library and use it to predict the class of a new news article.
 
10
  ---
11
 
12
  # Model Description
13
+ This is a fine-tuned DistilBART model for sequence classification on CNN news articles for text classification.
14
+ The model was fine-tuned using a batch size of 32,
15
+ a learning rate of 6e-5,
16
+ and for 1 epoch.
17
 
18
  ## Dataset
19
  The CNN News dataset was used for fine-tuning the model. The dataset consists of news articles from various categories such as sports, entertainment, politics, etc.
 
21
  ## Performance
22
  The following performance metrics were achieved after fine-tuning the model:
23
 
24
+ - Accuracy: 0.9597114707952147
25
+ - F1-score: 0.9589247895703302
26
+ - Recall: 0.9597114707952147
27
+ - Precision: 0.9589649408501851
28
 
29
  ## Usage
30
  You can use this model to classify CNN news articles into different categories such as sports, entertainment, politics, etc. You can load the model using the Hugging Face Transformers library and use it to predict the class of a new news article.