Chit-Chat simple basé sur KoGPT2

But

Expérience de chatbot simple utilisant les données de chatbot Hangul accessibles au public et KoGPT2 pré-entraîné
Nous explorons le potentiel de diverses utilisations de KoGPT2 et évaluons qualitativement ses performances.

Architecture

Bonjour ! convient aux données de formation. En appliquant GPT-2, nous avons conçu l'architecture suivante

dessin

Apprenez un modèle qui peut maximiser P(|, ) en mappant le champ Q des données à l'énoncé , le champ A à l'énoncé et l'étiquette d'émotion à Les étiquettes d’émotion suivent les définitions ici (occurrence quotidienne 0, rupture (négative) 1, amour (positif) 2).

Installer

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

Comment s'entraîner

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

Comment discuter !

 # CUDA_VISIBLE_DEVICES=0 python train.py --chatCUDA_VISIBLE_DEVICES=0 python train_torch.py --gpus 1 --chat
utilisateur > jackpot
 Simsimi > Tout ira bien.

utilisateur > Quel est votre nom ? Simsimi > Je suis Wirobot.

utilisateur > Que faisons-nous le jour de la fête des parents ? Simsimi > Joue avec moi.

utilisateur > Il pleut
Simsimi > Prenez un parapluie.

utilisateur > Il neige
Simsimi > Construisez un bonhomme de neige.

utilisateur > Le tonnerre frappe
Simsimi > Tu me manques aussi.

utilisateur > Le printemps est arrivé.
Simsimi > J'ai toujours hâte au printemps.

utilisateur > J'ai mal au ventre.
Simsimi > Rentrez chez vous et reposez-vous vite.

utilisateur > Qui vous a créé ? Simsimi > Vous êtes la personne qui vous a créé.