KoGPT2 chatbot

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

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

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

• เราสำรวจศักยภาพในการใช้งานต่างๆ ของ KoGPT2 และประเมินประสิทธิภาพในเชิงคุณภาพ

สถาปัตยกรรม

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

• เรียนรู้โมเดลที่สามารถเพิ่ม P(<sys>|<usr>, <sent>) ได้โดยการแมปฟิลด์ Q ของข้อมูลกับ <usr> คำพูด, ฟิลด์ A กับ <sys> คำพูด และป้ายกำกับอารมณ์ไปยัง <sent> ป้ายกำกับอารมณ์เป็นไปตามคำจำกัดความที่นี่ (เหตุการณ์ที่เกิดขึ้นทุกวัน 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 > คุณคือคนที่สร้างคุณ

บทแนะนำของ Colab

• ไพทอร์ช

•