rhaymison's picture
Update README.md
72ab179 verified
|
raw
history blame
7.61 kB
metadata
license: apache-2.0
base_model: google-t5/t5-small
tags:
  - generated_from_trainer
  - summarization
  - text2text
metrics:
  - rouge
model-index:
  - name: flan-t5-small-summarization
    results: []
inference:
  parameters:
    max_new_tokens: 128
pipeline_tag: summarization
datasets:
  - recogna-nlp/recognasumm
language:
  - pt
library_name: transformers
widget:
  - text: >-
      sumarize: Na segunda disputa estadual para escolher o candidato do partido
      republicano para as eleições de novembro nos Estados Unidos, o
      ex-presidente Donald Trump teve mais uma vitória. Ele venceu as primárias
      em New Hampshire, que ocorreram na terça-feira (23/01). Antes disso, o
      favoritismo de Trump, apontado por diversas pesquisas, foi visto em sua
      vitória na primeira prévia do calendário eleitoral americano, em Iowa.
      Naquele Estado, Trump registrou 51% dos votos e vantagem de 30 pontos
      sobre o segundo colocado, o governador da Flórida, Ron DeSantis. No
      domingo (21/1), DeSantis anunciou sua desistência da corrida presidencial
      de 2024 e manifestou apoio a Trump. O movimento deixou Nikki Haley,
      ex-embaixadora dos Estados Unidos nas Nações Unidas, como a única rival
      significativa de Trump no partido.
    example_title: Resumo 1
  - text: >-
      sumarize: No que consiste o transtorno dismórfico corporal? São pessoas
      que se acham feias e querem mudar sua aparência de forma obsessiva, mesmo
      que não tenham nenhum problema. Num dos estudos que fiz, detectamos que de
      50% a 54% dos pacientes que procuram cirurgia de face, nariz ou abdômen
      apresentam essa condição. A cirurgia pode beneficiar aqueles com um quadro
      leve ou intermediário do transtorno. No entanto, os que apresentam um
      transtorno mais grave não devem ser operados, e sim encaminhados para
      tratamento psicológico. A maior dificuldade é que aceitem ajuda. Muitos
      preferem buscar um médico que dê sinal verde para a intervenção.
    example_title: Resumo 2

t5-portuguese-small-summarization

Model description

This model aims to help supply the needs of models in the Portuguese language for certain tasks. The model presents a good performance for summary tasks. Some errors due to word accentuation may occasionally occur due to the small version of the model.

Card example 1 with t5-portuguese-small-summarization

Trump vence as primárias em New Hampshire, que ocorreram na terça-feira (23/01).
Antes disso, o favoritismo de Trump, apontado por diversas pesquisas, foi visto ao segundo colocado, governador da Flórida, Ron DeSantis.

Card example 1 with orginal t5-small

O movimento deixou Nikki Haley, ex-embaixadora dos Estados Unidos, como a nica rival significativa de Trump no partido.

This model is a fine-tuned version of google-t5/t5-small

  • Loss: 1.8997
  • Rouge1: 15.0817
  • Rouge2: 5.3292
  • Rougel: 12.958
  • Rougelsum: 13.8768
  • Gen Len: 18.968

GPU: RTX 3060, 12GB, =~3500 cuda cores

Training hyperparameters

The following hyperparameters were used during training:

  • learning_rate: 2e-05
  • train_batch_size: 6
  • eval_batch_size: 6
  • seed: 42
  • gradient_accumulation_steps: 4
  • total_train_batch_size: 24
  • optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
  • lr_scheduler_type: linear
  • num_epochs: 25
  • mixed_precision_training: Native AMP

Training results

Training Loss Epoch Step Validation Loss Rouge1 Rouge2 Rougel Rougelsum Gen Len
2.092 0.6 500 1.9551 14.6613 5.2159 12.5685 13.4544 18.968
2.092 0.72 600 1.9508 14.6862 5.2585 12.6345 13.5299 18.968
2.092 0.84 700 1.9473 14.7323 5.1636 12.6962 13.5118 18.968
2.092 0.96 800 1.9488 14.7104 5.1587 12.7019 13.5439 18.968
2.092 1.08 900 1.9397 14.8448 5.2826 12.7924 13.6464 18.968
2.077 1.2 1000 1.9373 14.9495 5.3975 12.8935 13.7491 18.968
2.077 1.32 1100 1.9372 14.93 5.4048 12.8809 13.7012 18.968
2.077 1.44 1200 1.9311 14.8196 5.2564 12.8279 13.6688 18.968
2.077 1.56 1300 1.9311 14.8757 5.2282 12.8286 13.7152 18.968
2.077 1.68 1400 1.9287 14.9308 5.3154 12.8522 13.7326 18.968
2.06 1.8 1500 1.9268 14.8923 5.2594 12.8387 13.6839 18.968
2.06 1.92 1600 1.9256 15.085 5.2911 12.9424 13.8375 18.968
2.06 2.04 1700 1.9245 14.9127 5.3024 12.8339 13.6987 18.968
2.06 2.16 1800 1.9197 15.0974 5.2812 12.9218 13.8758 18.968
2.06 2.28 1900 1.9172 15.0564 5.2437 12.8736 13.8318 18.968
2.0474 2.4 2000 1.9149 14.9414 5.1408 12.8381 13.7028 18.968
2.0474 2.52 2100 1.9149 15.0211 5.2195 12.954 13.809 18.968
2.0474 2.64 2200 1.9113 15.0689 5.2702 12.9338 13.8276 18.968
2.0474 2.76 2300 1.9129 15.134 5.2675 13.0113 13.9106 18.968
2.0474 2.88 2400 1.9103 15.1097 5.276 12.9856 13.8559 18.968
2.04 3.0 2500 1.9062 15.1413 5.2281 12.9537 13.8494 18.968
2.04 3.12 2600 1.9070 14.9792 5.2091 12.8586 13.695 18.968
2.04 3.24 2700 1.9066 14.9506 5.2238 12.8265 13.6925 18.968
2.04 3.36 2800 1.9063 15.053 5.2235 12.8833 13.7711 18.968
2.04 3.48 2900 1.9064 14.9386 5.1363 12.7915 13.688 18.968
2.0273 3.6 3000 1.9053 15.0901 5.2518 12.9063 13.8338 18.968
2.0273 3.72 3100 1.9059 15.0692 5.2665 12.932 13.8394 18.968
2.0273 3.84 3200 1.9021 15.0768 5.3179 12.9916 13.8653 18.968
2.0273 3.96 3300 1.9024 15.1808 5.3312 13.0143 13.9269 18.968
2.0273 4.08 3400 1.8981 15.0905 5.2769 12.9551 13.8666 18.968
2.0291 4.2 3500 1.9007 15.0453 5.3159 12.9429 13.824 18.968
2.0291 4.32 3600 1.9017 15.0403 5.3474 12.9625 13.8437 18.968
2.0291 4.44 3700 1.9005 15.0456 5.3468 12.9521 13.8413 18.968
2.0291 4.56 3800 1.8991 15.0501 5.3539 12.9597 13.8408 18.968
2.0291 4.68 3900 1.8998 15.1219 5.3599 12.9936 13.9013 18.968
2.0193 4.8 4000 1.9004 15.0831 5.329 12.9697 13.8762 18.968
2.0193 4.92 4100 1.8997 15.0817 5.3292 12.958 13.8768 18.968

Framework versions

  • Transformers 4.38.2
  • Pytorch 2.2.1+cu121
  • Datasets 2.18.0
  • Tokenizers 0.15.2