AyoubChLin
commited on
Commit
·
f94d9d0
1
Parent(s):
576d211
Update README.md
Browse files
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.
|
|
|
|
|
|
|
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.
|