Bate-papo simples baseado em KoGPT2

Propósito

Experimento simples de chatbot usando dados de chatbot Hangul disponíveis publicamente e KoGPT2 pré-treinado
Exploramos o potencial para vários usos do KoGPT2 e avaliamos qualitativamente seu desempenho.

Arquitetura

Olá! é adequado para dados de treinamento. Aplicando GPT-2, projetamos a seguinte arquitetura

desenho

Aprenda um modelo que pode maximizar P(|, ) mapeando o campo Q dos dados para a expressão , o campo A para a expressão e o rótulo de emoção para Os rótulos de emoções seguem as definições aqui (ocorrência diária 0, rompimento (negativo) 1, amor (positivo) 2).

Instalar

 clone git --recurse-submodules https://github.com/haven-jeon/KoGPT2-chatbot.gitcd KoGPT2-chatbot
pip3 instalar -r requisitos.txt

Como treinar

 CUDA_VISIBLE_DEVICES=0 python train_torch.py --gpus 1 --train --max_epochs 2

Como conversar!

 # CUDA_VISIBLE_DEVICES=0 python train.py --chatCUDA_VISIBLE_DEVICES=0 python train_torch.py --gpus 1 --chat
usuário > jackpot
 Simsimi > Vai ficar tudo bem.

usuário > Qual é o seu nome? Simsimi > Eu sou Wirobot.

usuário > O que fazemos no Dia dos Pais Simsimi > Brinque comigo.

usuário > Está chovendo
Simsimi > Leve um guarda-chuva.

usuário > Está nevando
Simsimi > Construa um boneco de neve.

usuário > Trovões
Simsimi > Também sinto sua falta.

usuário > A primavera chegou.
Simsimi > Estou sempre animado com a primavera.

usuário > Estou com dor de estômago.
Simsimi > Vá para casa e descanse rapidamente.

usuário > Quem fez você? Simsimi > Você é a pessoa que fez você.