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 > 我是 Wirobot。

用户 > 父母节我们做什么？ > 和我一起玩。

用户 > 下雨了
Simsimi > 带把伞。

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

用户 > 雷霆来袭
西西米 > 我也想你。

用户 > 春天来了。
Simsimi > 我总是对春天感到兴奋。

用户 > 我胃不舒服。
Simsimi > 赶紧回家休息吧。

user > 谁创造了你？Simsimi > 你是创造你的人。

Colab 教程

• 火炬

•