PamelaBorelli
commited on
Commit
•
e8d1a30
1
Parent(s):
1d36027
Create README.md
Browse files
README.md
ADDED
@@ -0,0 +1,71 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
# Model Card para PamelaBorelli/flan-t5-base-summarization-pt-br
|
2 |
+
|
3 |
+
## Dados Gerais
|
4 |
+
|
5 |
+
- **Nome:** [PamelaBorelli/flan-t5-base-summarization-pt-br](PamelaBorelli/flan-t5-base-summarization-pt-br)
|
6 |
+
- **Tipo:** Languege Model, Transformer Encoder-Decoder
|
7 |
+
- **Licença:** Language model
|
8 |
+
- **Modelo base:** [google/flan-t5-base](https://huggingface.co/google/flan-t5-base#model-details)
|
9 |
+
- **Modelo Relacionado:** Checkpoints do FLAN-T5
|
10 |
+
- **Checkpoints Originais:** Checkpoints do FLAN-T5
|
11 |
+
|
12 |
+
## Resumo
|
13 |
+
|
14 |
+
O modelo original que serviu de base para o modelo final é o [flan-t5-base](https://huggingface.co/google/flan-t5-base#model-details). Um modelo multilinguagem com tamanho de 248M e que possui uma arquitetura baseada no T5 (Text-to-Text Transfer Transformer) com codificador-decodificador. O Flan-T5 original foi ajustado em uma mistura de tarefas com objetivo de melhorar sua capacidade de generalização.
|
15 |
+
|
16 |
+
O modelo final [PamelaBorelli/flan-t5-base-summarization-pt-br](PamelaBorelli/flan-t5-base-summarization-pt-br) foi treinado no processo de **instrução e ajuste fino** (*finetune*). Foi realizado dois *finetunes*, primeiro para tradução de textos usando o dataset e posteriormente para tarefa de sumarização com datasets ambos em Português-BR.
|
17 |
+
|
18 |
+
## Utilização Pretendida
|
19 |
+
|
20 |
+
O modelo foi trabalhado para tarefa final de sumarização de textos em Português-BR. Não foi testado para outros idiomas.
|
21 |
+
|
22 |
+
### Uso
|
23 |
+
|
24 |
+
```
|
25 |
+
from transformers import T5Tokenizer, T5ForConditionalGeneration
|
26 |
+
|
27 |
+
tokenizer = T5Tokenizer.from_pretrained("PamelaBorelli/flan-t5-base-summarization-pt-br")
|
28 |
+
model = T5ForConditionalGeneration.from_pretrained("PamelaBorelli/flan-t5-base-summarization-pt-br")
|
29 |
+
|
30 |
+
input_text = "O corpo está mais propenso a sentir dores com exercícios de alta intensidade | Foto: Getty Images O problema está em saber identificar qual é qual. "Em algumas situações, é difícil diferenciar uma da outra", reconhece Juan Francisco Marco, professor do Centro de Ciência do Esporte, Treinamento e Fitness Alto Rendimento, na Espanha. "A dor boa é aquela que associamos ao exercício físico, que não limita (o movimento) e permite continuar (a se exercitar) até o momento em que o músculo fica realmente esgotado e não trabalha mais", explica. É importante detectar qual é o tipo de dor que você está sentindo, para evitar ter problemas mais sérios | Foto: Getty Images Para Francisco Sánchez Diego, diretor do centro de treinamento Corpore 10, "a dor boa se sente no grupo muscular que você trabalhou, tanto durante o treinamento como nos dias seguintes"."
|
31 |
+
input_ids = tokenizer(input_text, return_tensors="pt").input_ids
|
32 |
+
|
33 |
+
outputs = model.generate(input_ids)
|
34 |
+
print(tokenizer.decode(outputs[0]))
|
35 |
+
```
|
36 |
+
|
37 |
+
|
38 |
+
## Idiomas
|
39 |
+
|
40 |
+
Português-BR
|
41 |
+
|
42 |
+
## Dados de Treinamento
|
43 |
+
|
44 |
+
O modelo foi treinado para tarefa de sumarização usando os parâmetros:
|
45 |
+
|
46 |
+
```
|
47 |
+
evaluation_strategy="steps" #forma para avaliar o output
|
48 |
+
eval_steps= #quantidade de steps para avaliar o output
|
49 |
+
learning_rate= #taxa de aprendizado
|
50 |
+
per_device_train_batch_size= #tamanho do batch para o treinamento
|
51 |
+
per_device_eval_batch_size= #tamanho do batch para a validação
|
52 |
+
gradient_accumulation_steps= #até quanto deve acumular os batchs
|
53 |
+
weight_decay= #regularização L2
|
54 |
+
num_train_epochs= #quantidade de épocas para treinamento
|
55 |
+
save_strategy="steps" #forma para salvar o output
|
56 |
+
save_steps = #quantidade de steps para salvar o output
|
57 |
+
push_to_hub=False #salvar o modelo no hub do Hugging Face
|
58 |
+
load_best_model_at_end=True #carregar o melhor modelo no final do treinamento (necessário para o callback)
|
59 |
+
```
|
60 |
+
|
61 |
+
Para tokenização foi usado os parâmetros:
|
62 |
+
|
63 |
+
```
|
64 |
+
start_prompt= "Sumarize: \n" #início instrução para sumarização
|
65 |
+
end_prompt= "\n\nTradução: " #fim da instrução para sumarização
|
66 |
+
input_name="coluna_imput" #nome da coluna no dataset com os textos de origem
|
67 |
+
target_name="coluna_target" #nome da coluna no dataset com os textos de destino
|
68 |
+
max_input_length = 256 #tamanho máximo de input para tokenizar
|
69 |
+
max_target_length = 256 #tamanho máximo de target para tokenizar
|
70 |
+
columns_to_remove= ['coluna_to_remove'] #colunas que serão removidas do dataset original
|
71 |
+
```
|