聊天机器人

聊天机器人的 PyTorch 实现。

依赖

+- Python 3.7 +- PyTorch 1.0 +- jieba 0.39 +- tqdm 4.31.1

数据集

小黄鸡聊天语料 PTT中文語料

用法

数据预处理

提取训练数据：

$ python extract.py
$ python pre_process.py

训练

$ python train.py

可视化训练过程，执行：

$ tensorboard --logdir path_to_current_dir/logs

演示

$ python demo.py

不正经的语料只能训练不正经的聊天机器人 ╮（﹀_﹀）╭