Chit-Chat อย่างง่ายบนพื้นฐานของ KoGPT2

วัตถุประสงค์

การทดลองแชทบอทอย่างง่ายโดยใช้ข้อมูลแชทบอทอังกูลที่เปิดเผยต่อสาธารณะและ KoGPT2 ที่ผ่านการฝึกอบรมล่วงหน้า
เราสำรวจศักยภาพในการใช้งานต่างๆ ของ KoGPT2 และประเมินประสิทธิภาพในเชิงคุณภาพ

สถาปัตยกรรม

Hello! เหมาะสำหรับข้อมูลการฝึกอบรม เมื่อใช้ GPT-2 เราได้ออกแบบสถาปัตยกรรมต่อไปนี้

การวาดภาพ

เรียนรู้โมเดลที่สามารถเพิ่ม P(|, ) ได้โดยการแมปฟิลด์ Q ของข้อมูลกับ คำพูด, ฟิลด์ A กับ คำพูด และป้ายกำกับอารมณ์ไปยัง ป้ายกำกับอารมณ์เป็นไปตามคำจำกัดความที่นี่ (เหตุการณ์ที่เกิดขึ้นทุกวัน 0, การเลิกรา (เชิงลบ) 1, ความรัก (เชิงบวก) 2)

ติดตั้ง

 git clone --recurse-submodules https://github.com/haven-jeon/KoGPT2-chatbot.gitcd KoGPT2-chatbot
pip3 ติดตั้ง -r ข้อกำหนด txt

วิธีการฝึก

 CUDA_VISIBLE_DEVICES=0 หลาม train_torch.py --gpus 1 --train --max_epochs 2

แชทยังไง!

 # CUDA_VISIBLE_DEVICES=0 หลาม train.py --chatCUDA_VISIBLE_DEVICES=0 หลาม train_torch.py --gpus 1 --chat
ผู้ใช้ > แจ็คพอต
 ซิมซิมิ > มันจะไม่เป็นไร

ผู้ใช้ > คุณชื่ออะไรซิมซิมิ > ฉันชื่อไวโรบอต

ผู้ใช้ > เราทำอะไรในวันพ่อแม่? Simsimi > เล่นกับฉัน

ผู้ใช้ > ฝนตก
Simsimi > กางร่ม

ผู้ใช้ > หิมะตก
Simsimi > สร้างตุ๊กตาหิมะ

ผู้ใช้ > สายฟ้าฟาด
ซิมซิมิ > ฉันก็คิดถึงคุณเหมือนกัน

ผู้ใช้ > ฤดูใบไม้ผลิมาแล้ว
Simsimi > ฉันตื่นเต้นกับฤดูใบไม้ผลิอยู่เสมอ

ผู้ใช้ > ฉันท้องเสีย
Simsimi > กลับบ้านและพักผ่อนอย่างรวดเร็ว

ผู้ใช้ > ใครสร้างคุณ Simsimi > คุณคือคนที่สร้างคุณ