KoGPT2 聊天機器人

基於 KoGPT2 的簡單閒聊

目的

• 使用公開的 Hangul 聊天機器人資料和預先訓練的 KoGPT2 進行簡單的聊天機器人實驗

• 我們探索KoGPT2各種用途的潛力並定性評估其性能。

建築學

你好！應用GPT-2，我們設計了以下架構

• 透過將資料的 Q 欄位對應到話語、將 A 欄位對應到話語、將情緒標籤對應到學習可以最大化P(|, )的模型情緒標籤遵循此處的定義（日常發生 0、分手（負）1、愛（正向）2）。

安裝

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

如何訓練

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

如何聊天！

 # CUDA_VISIBLE_DEVICES=0 python train.py --chatCUDA_VISIBLE_DEVICES=0 python train_torch.py​​ --gpus 1 --chat
用戶 > 大獎
 辛西米 > 會沒事的。

使用者 > 你叫什麼名字？

使用者 > 父母節我們做什麼？

用戶 > 下雨了
Simsimi > 附傘。

用戶 > 下雪了
Simsimi > 堆雪人。

用戶 > 雷霆來襲
西西米 > 我也想你。

使用者 > 春天來了。
Simsimi > 我總是對春天感到興奮。

使用者 > 我胃不舒服。
Simsimi > 趕快回家休息。

user > 誰創造了你？

Colab 教程

• 火炬

•