Este projeto consiste do desenvolvimento de um chatbot para busca e marcação de passagens áreas.
- Classificadores de intenção, extratores de entidade e interface de chat utilizando deep learning encontram-se no notebook
PassagensChatbot.ipynb
- Projeto de chatbot no rasa encontra-se no diretório
rasa_chatbot
- Apresentação no Canva
- Apresentação no formato PDF encontra-se no arquivo
PassagensChatbot.pdf
- Apresentação em vídeo
Os dados utilizados para treinamento do chatbot encontram-se disponíveis no dstc8 schema guided dialogue. Foi selecionado um subconjunto deste conjunto de dados para refletir apenas o domínio de viagem.
As intenções que serão classficadas pelo chatbot serão:
- ReserveOnewayFlight
- SearchOnewayFlight
- SearchRoundtripFlights
- ReserveRoundtripFlights
- NONE *verificar se isso é válido
O tamanho do sobconjunto de dados definido é uma característica muito relevante para a tarefa de classificação da intenção, tendo em vista que a pequena quantidade de dados de treinamento poderá resultar em overfitting dos modelos de aprendizagem.
Para a tarefa de classificação de intenções, realizamos uma análise comparativa dos resultados obtidos durante treinamento e teste dos seguintes modelos:
- LSTM
- CNN
- SVM
- Transformer (BERT)
Também foram implementados extratores de entidades para reconhecimento de entidades da sentenças baseados em transformer (BERT) e LSTM.
A base de dados sob estudo foi previamente dividida em subconjuntos de treinamento e de teste, de forma que ambas amostras nunca se misturam durante os processos de treino e teste. A avaliação da performance dos modelos sob o subcojunto de teste mede a acurácia e o F1-score dos modelos treinados anteriormente.
[1] Rastogi, Abhinav, Xiaoxue, Zang, Srinivas, Sunkara, Raghav, Gupta, Pranav, Khaitan. "Towards scalable multi-domain conversational agents: The schema-guided dialogue dataset." Proceedings of the AAAI Conference on Artificial Intelligence. 2020.
[2] Lee, Harrison, Raghav, Gupta, Abhinav, Rastogi, Yuan, Cao, Bin, Zhang, Yonghui, Wu. "SGD-X: A Benchmark for Robust Generalization in Schema-Guided Dialogue Systems." Proceedings of the AAAI Conference on Artificial Intelligence. 2022.
[4] Rasa FAQs