Obrolan Sederhana berdasarkan KoGPT2

Tujuan

Eksperimen chatbot sederhana menggunakan data chatbot Hangul yang tersedia untuk umum dan KoGPT2 yang telah dilatih sebelumnya
Kami mengeksplorasi potensi berbagai penggunaan KoGPT2 dan mengevaluasi kinerjanya secara kualitatif.

Arsitektur

Halo! cocok untuk data pelatihan. Menerapkan GPT-2, kami merancang arsitektur berikut

menggambar

Pelajari model yang dapat memaksimalkan P(|, ) dengan memetakan bidang Q data ke ucapan , bidang A ke ucapan , dan label emosi ke Label emosi mengikuti definisi di sini (kejadian sehari-hari 0, putus cinta (negatif) 1, cinta (positif) 2).

Memasang

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

Cara Melatih

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

Bagaimana Mengobrol!

 # CUDA_VISIBLE_DEVICES=0 python train.py --chatCUDA_VISIBLE_DEVICES=0 python train_torch.py --gpus 1 --chat
pengguna > jackpot
 Simsimi > Ini akan baik-baik saja.

pengguna > Siapa nama Anda? Simsimi > Saya Wirobot.

pengguna > Apa yang kita lakukan di Hari Orang Tua? Simsimi > Bermain dengan saya.

pengguna > Sedang hujan
Simsimi > Ambil payung.

pengguna > Sedang turun salju
Simsimi > Bangun manusia salju.

pengguna > Guntur menyambar
Simsimi > Aku juga merindukanmu.

pengguna > Musim semi telah tiba.
Simsimi > Saya selalu bersemangat dengan musim semi.

pengguna > Saya sakit perut.
Simsimi > Cepat pulang dan istirahat.

pengguna > Siapa yang menciptakan Anda? Simsimi > Anda adalah orang yang menciptakan Anda.