• Stars
    star
    163
  • Rank 231,141 (Top 5 %)
  • Language
    Python
  • License
    MIT License
  • Created almost 7 years ago
  • Updated over 1 year ago

Reviews

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

Repository Details

A dumb chat bot implemented using PyTorch and trained with Cornell Movie Dialogue

DUMB chatbot

DUMB(Dumb Undereducated Maladroit Bot) chatbot, a chatbot implemented with PyTorch and trained with Cornell Movie Dialogs Corpus.

CAUTION

Under current support situation of PyTorch, MPS devices (commonly known as Apple Silicon GPU) suffer from extreme slow training, so just don't train with them now.

TODO:

  • Breakpoint continuous training to achieve intermittent training.
  • Migrate to Python3.

Requirements

  • Python 3.10
  • PyTorch 1.13.1
  • torchaudio 0.13.1
  • torchvision 0.14.1
  • festival (Linux Environment)
  • say (macOS Environment)

Training Resource

How to use

Data Laundry

python3 prerequisites.py

The script would create dialogue_corpus.txt under ./data.

Model Training

python3 train.py

Configs are stored in config.json. Model Training could be time-consuming. I would strongly recommend enabling CUDA in config.json to accelerate the whole training process.

{
  "TRAIN": {
    "DEVICE": "cuda",
    ...
  }
}

And if you are using Apple Silicon GPUs, do the following:

{
  "TRAIN": {
    "DEVICE": "mps",
    ...
  }
}

Testing and Running

python3 chatbot.py

Test Samples

> hi .
bot: hi .
> what is your name ?
bot: vector frankenstein .
> how are you ?
bot: fine .
> where are you from ?
bot: up north .
> are you happy today ?
bot: yes .

Though it could answer some questions, it's still dumb enough.

References

Made with ❤ by Justin Lee!

™ and © 1997-2022 Justin Lee. All Rights Reserved. License Agreement

More Repositories

1

Fuck-Tencent

作为一个学习党,你还总是被室友的打游戏的机械键盘声音吵到么,你还总是被室友在游戏里要被杀死的绝望叫声所震撼么,你还被室友输了游戏以后久久不能释怀的喃喃自语所心烦么?不要犹豫了,点进来!点一下,你点不到吃亏;点一下,你点不到上当!来吧朋友,让我们还你一片没有小学生的蓝天!
Python
178
star
2

LiMe

LiMe is a communication application implemented in Java that allows you to keep in touch and exchange files with friends anytime, anywhere. You can also quickly deploy a custom LiMe server for private communication.
Java
114
star
3

FloweringTree

Plain, old and simple Flowering Tree Class
Python
112
star
4

LiAg

LiAg (LiXin Avatar Generator) is an open source 3D avatar modeling software implemented in React.js and WebGL, which provides web side 3D avatar modeling and rendering service, and users can export the avatars into STL files which can be used for 3D printing.
JavaScript
111
star
5

Library-Management

数据库大作业——图书馆管理系统前端:BootCamp+JQuery 后端:Spring
JavaScript
110
star
6

LiXinCompiler

参考自龙书上的语言,自己实现了一个编译器的前端。
Java
101
star
7

weather-dashboard

ReactJS based weather dashboard runs on my RSBP
JavaScript
79
star
8

iMelody

A social Beat Sequencer through which you can share your melodies with your friends.
Java
71
star
9

LiCMS

LiCMS (Lixin Content Management System) is a content management system used for blogging, implemented with Flask.
Python
17
star
10

Super-Basic-C-stuff

大一的C语言课程,基础算法示例。
C
7
star
11

Fake-Tencent

A Instant Messaging Software not intended to take the place of Tencnent QQ or WeChat
Swift
5
star
12

realJustinLee.github.io

This is the source code of my presonal blog. And here goes the link ->
HTML
5
star
13

Cryption

加密解密与算法测试
C#
5
star
14

Computer-Network

<钱权老师的网络课> 的各种研讨作业
C
4
star
15

MFC-Sales-System

这两天临时起义拿起MFC随意写了点东西
C++
4
star
16

nester_lixin

A simple printer of nested lists
Python
4
star
17

FakeEleme

安卓大作业
Java
4
star
18

Concentration

The implementation of the Concentration Game in CS193p Stanford
Swift
4
star
19

luci-theme-fruitbingo

A theme for HomeLede and Openwrt.
CSS
3
star
20

Game_of_Life

John Conway, inventor of the Game of Life, has died of COVID-19. In his honor I'm making this Game of Life project.
JavaScript
2
star
21

realJustinLee

1
star
22

dayz-server-manager

Slightly modded dayz server manager to ease my usage
PowerShell
1
star
23

LiUPS

Lixin Universal Proxy Server
Shell
1
star
24

markdown_resume

My MarkDown resume
1
star