https://github.com/matteopolak/chatterbox

RNN-based chat message generator.
https://github.com/matteopolak/chatterbox

numpy pandas python rnn tensorflow text-generation

Last synced: 12 days ago
JSON representation

RNN-based chat message generator.

• Host: GitHub
• URL: https://github.com/matteopolak/chatterbox
• Owner: matteopolak
• License: mit
• Created: 2022-11-20T01:25:46.000Z (almost 2 years ago)
• Default Branch: main
• Last Pushed: 2023-11-04T19:22:25.000Z (about 1 year ago)
• Last Synced: 2023-11-04T20:25:17.968Z (about 1 year ago)
• Topics: numpy, pandas, python, rnn, tensorflow, text-generation
• Language: Python
• Homepage:
• Size: 10.8 MB
• Stars: 0
• Watchers: 3
• Forks: 0
• Open Issues: 0
• Metadata Files:
  • Readme: README.md
  • License: LICENSE

Awesome Lists containing this project

README

        
# Chat Generator

![Build Status](https://github.com/matteopolak/chatterbox/actions/workflows/yapf.yml/badge.svg)

[![License:MIT](https://img.shields.io/badge/license-MIT-yellow.svg)](https://opensource.org/licenses/MIT)

## Training on new text

```bash

python src/train.py --model atrioc --data data/atrioc.csv

```

### Chat dataset format

The message dataset must be a CSV file where the first column is the username of the chatter, and the second column is the message content. The column header should not be present. For example:

```csv

mattepolak,"Hello, world!"

```

### Using a GPU

To use a GPU, [follow the official instructions](https://www.tensorflow.org/install/pip#step-by-step_instructions).

## Generating text

```bash

python src/generate.py --model atrioc --length 500 --text "atrioc: i love"

```