fabiochiu commited on
Commit
7b6462c
1 Parent(s): f673e66

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +9 -20
README.md CHANGED
@@ -9,34 +9,23 @@ model-index:
9
  <!-- This model card has been generated automatically according to the information Keras had access to. You should
10
  probably proofread and complete it, then remove this comment. -->
11
 
12
- # t5-base-medium-title-generation
13
 
14
- This model is (t5-base)[https://huggingface.co/t5-base] fine-tuned on the 190k Medium Articles dataset for predicting article titles using the article textual content as input.
15
-
16
-
17
- ## Model description
18
-
19
- More information needed
20
-
21
- ## Intended uses & limitations
22
-
23
- More information needed
24
 
25
  ## Training and evaluation data
26
 
27
- More information needed
28
-
29
- ## Training procedure
30
-
31
- ### Training hyperparameters
32
-
33
- The following hyperparameters were used during training:
34
- - optimizer: None
35
- - training_precision: float32
36
 
37
  ### Training results
38
 
 
39
 
 
 
 
 
 
40
 
41
  ### Framework versions
42
 
 
9
  <!-- This model card has been generated automatically according to the information Keras had access to. You should
10
  probably proofread and complete it, then remove this comment. -->
11
 
12
+ # Model description
13
 
14
+ This model is [t5-base](https://huggingface.co/t5-base) fine-tuned on the [190k Medium Articles](https://www.kaggle.com/datasets/fabiochiusano/medium-articles) dataset for predicting article titles using the article textual content as input.
 
 
 
 
 
 
 
 
 
15
 
16
  ## Training and evaluation data
17
 
18
+ The model has been trained on a single epoch spanning about 16000 articles, evaluating on 1000 random articles not used during training.
 
 
 
 
 
 
 
 
19
 
20
  ### Training results
21
 
22
+ The model has been evaluated on a random dataset split of 1000 articles not used during training and validation.
23
 
24
+ - Rouge-1: 33.8%
25
+ - Rouge-2: 17.7%
26
+ - Rouge-L: 31.4%
27
+ - Rouge-Lsum: 31.4%
28
+ - Average length of the generated titles: 12 tokens
29
 
30
  ### Framework versions
31