روبوت الدردشة KoGPT2

دردشة بسيطة تعتمد على KoGPT2

غاية

• تجربة chatbot بسيطة باستخدام بيانات Hangul chatbot المتاحة للعامة وKoGPT2 المدربة مسبقًا

• نحن نستكشف إمكانات الاستخدامات المختلفة لـ KoGPT2 ونقوم بتقييم أدائها نوعيًا.

بنيان

مرحبا! مناسب لبيانات التدريب. بتطبيق 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 > أنا Wirobot.

المستخدم > ماذا سنفعل في يوم الوالدين؟ العب معي؟

المستخدم > إنها تمطر
Simsimi > خذ مظلة.

المستخدم > إنها تثلج
Simsimi > قم ببناء رجل ثلج.

المستخدم > ضربات الرعد
Simsimi > اشتقت لك أيضا.

المستخدم > لقد حان الربيع.
Simsimi > أنا دائما متحمسة لفصل الربيع.

المستخدم > لدي اضطراب في المعدة.
Simsimi > عد إلى المنزل واسترح بسرعة.

المستخدم > من صنعك؟ Simsimi > أنت الشخص الذي صنعك.

دروس كولاب

• باي تورش

•