import numpy as np
import pandas as pd
from datasets import load_dataset,Dataset
from transformers import AutoTokenizer, DataCollatorWithPadding
from transformers import TrainingArguments
from transformers import AutoModelForSequenceClassification
from transformers import Trainer
from datasets import load_metric
train = pd.read_csv('treino_bert_filter.csv')
teste = pd.read_csv('teste_bert_filter.csv')
def encode(x):
if x == 2:
return 1
else:
return 0
train['label'] = train['label'].apply(encode)
teste['label'] = teste['label'].apply(encode)
dataset_train = Dataset.from_pandas(train)
dataset_teste = Dataset.from_pandas(teste)
Um tokenizer está encarregado de preparar as entradas para um modelo. Portanto, nessa etapa o objetivo é transformar o pedido e reposta em vetores de embeddings, para esse processo se faz necessário um modelo treinado em uma base com domínio em português para que todas as palavras presentes na nossa base consiga se adaptar ao processo, desse modo, para essa etapa usamos o modelo pierreguillou/bert-base-cased-squad-v1.1-portuguese disponibilizado gratuitamente na plataforma Hugginface.
checkpoint = "pierreguillou/bert-base-cased-squad-v1.1-portuguese"
tokenizer = AutoTokenizer.from_pretrained(checkpoint)
def tokenize_function(example):
return tokenizer(example["sentence1"],example["sentence2"], truncation=True,max_length=512)
tokenized_train = dataset_train.map(tokenize_function, batched=True)
tokenized_test = dataset_teste.map(tokenize_function, batched=True)
data_collator = DataCollatorWithPadding(tokenizer=tokenizer)
Nessa etapa iremos utilizar a métrica Acurácia como refecia para avaliar nosso modelo, lembrando que:
No nosso contexto, dado que a variável de interesse é o atendimento do pedido, temos:
Falso Positivo (FP): O modelo previu que o pedido foi atendido, entretanto, não foi.
Falso Negativo (FN): O modelo previu que o pedido não foi atendido, entretanto, ele foi.
Verdadeiro Negativo (VN): O modelo previu que o pedido não foi atendido, e de fato ele não foi.
Verdadeiro Positivo (VP): O modelo previu que o pedido foi atendido, e de fato ele foi.
VP + VN / (VP + FP + VN + FN)
result = []
def compute_metrics(eval_preds):
metric = load_metric("accuracy")
logits, labels = eval_preds
predictions = np.argmax(logits, axis=-1)
result.append(predictions)
return metric.compute(predictions=predictions, references=labels)
Iremos utilizar um modelo de arquitetura BERT para a classificação das 3 classes, para entender melhor:
BERT (Bidirectional Encoder Representations from Transformers) é um artigo recente publicado por pesquisadores do Google AI Language. Isso causou um rebuliço na comunidade de aprendizado de máquina ao apresentar resultados de última geração em uma ampla variedade de tarefas de NLP, incluindo resposta a perguntas (SQuAD v1.1), inferência de linguagem natural (MNLI) e outras. A principal inovação técnica do BERT é a aplicação do treinamento bidirecional do Transformer, um modelo de atenção popular, à modelagem de linguagem. Isso está em contraste com os esforços anteriores que olhavam para uma sequência de texto da esquerda para a direita ou treinamento combinado da esquerda para a direita e da direita para a esquerda. Os resultados do artigo mostram que um modelo de linguagem que é treinado bidirecionalmente pode ter um senso mais profundo de contexto e fluxo de linguagem do que modelos de linguagem de direção única. No artigo, os pesquisadores detalham uma nova técnica chamada Masked LM (MLM) que permite o treinamento bidirecional em modelos nos quais era impossível anteriormente
Para realizar o treinamento em cima desse modelo realizamos o processo de adaptação ao nosso problema de classificação das três classes e usando como entrada o pedido e resposta ao pedido para que desse modo o modelo gerado conseguisse classificar entre Não Atendido e Atendido, para isso realizamos o processo de Transfer Learning.
O Transfer Learning é uma técnica em que um modelo de aprendizado profundo treinado em um grande conjunto de dados é usado para realizar tarefas semelhantes em outro conjunto de dados. Chamamos esse modelo de aprendizado profundo de modelo pré-treinado. Os exemplos mais renomados de modelos pré-treinados são os modelos de aprendizado profundo de visão computacional treinados no conjunto de dados ImageNet . Portanto, é melhor usar um modelo pré-treinado como ponto de partida para resolver um problema, em vez de construir um modelo do zero. Desse modo, para o nosso problema é melhor reutilizar um modelo com amplo domínio da língua portuguesa e consequentemente ajuda no processo de entediamento geral das sentenças.
training_args = TrainingArguments("test-trainer", evaluation_strategy="epoch")
training_args.report_to = []
model = AutoModelForSequenceClassification.from_pretrained("viniaraujoo/bert_transparencia_brasil", num_labels=3)
trainer = Trainer(
model,
training_args,
train_dataset=tokenized_train,
eval_dataset=tokenized_test,
data_collator=data_collator,
tokenizer=tokenizer,
compute_metrics=compute_metrics
)
%%time
trainer.train()
The following columns in the training set don't have a corresponding argument in `BertForSequenceClassification.forward` and have been ignored: sentence2, Unnamed: 0, sentence1. ***** Running training ***** Num examples = 6899 Num Epochs = 3 Instantaneous batch size per device = 8 Total train batch size (w. parallel, distributed & accumulation) = 8 Gradient Accumulation steps = 1 Total optimization steps = 2589
| Epoch | Training Loss | Validation Loss | Accuracy |
|---|---|---|---|
| 1 | 0.136200 | 0.935629 | 0.851995 |
| 2 | 0.040200 | 0.830477 | 0.878378 |
| 3 | 0.019900 | 0.781871 | 0.875804 |
Saving model checkpoint to test-trainer\checkpoint-500 Configuration saved in test-trainer\checkpoint-500\config.json Model weights saved in test-trainer\checkpoint-500\pytorch_model.bin tokenizer config file saved in test-trainer\checkpoint-500\tokenizer_config.json Special tokens file saved in test-trainer\checkpoint-500\special_tokens_map.json The following columns in the evaluation set don't have a corresponding argument in `BertForSequenceClassification.forward` and have been ignored: sentence2, Unnamed: 0, sentence1. ***** Running Evaluation ***** Num examples = 1554 Batch size = 8 Saving model checkpoint to test-trainer\checkpoint-1000 Configuration saved in test-trainer\checkpoint-1000\config.json Model weights saved in test-trainer\checkpoint-1000\pytorch_model.bin tokenizer config file saved in test-trainer\checkpoint-1000\tokenizer_config.json Special tokens file saved in test-trainer\checkpoint-1000\special_tokens_map.json Saving model checkpoint to test-trainer\checkpoint-1500 Configuration saved in test-trainer\checkpoint-1500\config.json Model weights saved in test-trainer\checkpoint-1500\pytorch_model.bin tokenizer config file saved in test-trainer\checkpoint-1500\tokenizer_config.json Special tokens file saved in test-trainer\checkpoint-1500\special_tokens_map.json The following columns in the evaluation set don't have a corresponding argument in `BertForSequenceClassification.forward` and have been ignored: sentence2, Unnamed: 0, sentence1. ***** Running Evaluation ***** Num examples = 1554 Batch size = 8 Saving model checkpoint to test-trainer\checkpoint-2000 Configuration saved in test-trainer\checkpoint-2000\config.json Model weights saved in test-trainer\checkpoint-2000\pytorch_model.bin tokenizer config file saved in test-trainer\checkpoint-2000\tokenizer_config.json Special tokens file saved in test-trainer\checkpoint-2000\special_tokens_map.json Saving model checkpoint to test-trainer\checkpoint-2500 Configuration saved in test-trainer\checkpoint-2500\config.json Model weights saved in test-trainer\checkpoint-2500\pytorch_model.bin tokenizer config file saved in test-trainer\checkpoint-2500\tokenizer_config.json Special tokens file saved in test-trainer\checkpoint-2500\special_tokens_map.json The following columns in the evaluation set don't have a corresponding argument in `BertForSequenceClassification.forward` and have been ignored: sentence2, Unnamed: 0, sentence1. ***** Running Evaluation ***** Num examples = 1554 Batch size = 8 Training completed. Do not forget to share your model on huggingface.co/models =)
Wall time: 13min 8s
TrainOutput(global_step=2589, training_loss=0.05833678047092507, metrics={'train_runtime': 788.9425, 'train_samples_per_second': 26.234, 'train_steps_per_second': 3.282, 'total_flos': 5165725772096190.0, 'train_loss': 0.05833678047092507, 'epoch': 3.0})
Vamos avaliar como está o comportamento do nosso modelo, em relação á métrica de acurácia e também a como o modelo está se comportando em relação a cada uma das classes de classificação, analisar o processo de distribuição de predição entre as classes para podermos considerar a real utilidade do modelo.
trainer.evaluate()
The following columns in the evaluation set don't have a corresponding argument in `BertForSequenceClassification.forward` and have been ignored: sentence2, Unnamed: 0, sentence1. ***** Running Evaluation ***** Num examples = 1554 Batch size = 8
{'eval_loss': 0.7818707227706909,
'eval_accuracy': 0.8758043758043758,
'eval_runtime': 19.4201,
'eval_samples_per_second': 80.02,
'eval_steps_per_second': 10.041,
'epoch': 3.0}
predictions_labels = result[1]
from ml_things import plot_dict, plot_confusion_matrix, fix_text
plot_confusion_matrix(y_true=teste['label'], y_pred=predictions_labels,
classes=['Nao Atendido','Atendido'], normalize=True,
magnify=0.1,
);
Normalized confusion matrix
Considerando o processo que o nosso modelo foi ótimo, iremos subir essa versão 1.1 para o repositório tbrasil da ong transparência Brasil para que posa ser utilizado, nessa etapa usamos a plataforma Hugginface para auxiliar.
trainer.model.push_to_hub("tbrasil/classificador_de_atendimento_2_classes_v1.1")
C:\Users\LMD\anaconda3\lib\site-packages\ipykernel\ipkernel.py:287: DeprecationWarning: `should_run_async` will not call `transform_cell` automatically in the future. Please pass the result to `transformed_cell` argument and any exception that happen during thetransform in `preprocessing_exc_tuple` in IPython 7.17 and above. and should_run_async(code) Configuration saved in tbrasil/classificador_de_atendimento_2_classes_v1.1\config.json Model weights saved in tbrasil/classificador_de_atendimento_2_classes_v1.1\pytorch_model.bin
'https://huggingface.co/tbrasil/classificador_de_atendimento_2_classes_v1.1/commit/147ae7455fb7891fbcef6e27de67badb01055d22'