PamelaBorelli commited on
Commit
e8d1a30
1 Parent(s): 1d36027

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +71 -0
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
+ ```