All Projects → wrannaman → tensorflow-pickup-lines

wrannaman / tensorflow-pickup-lines

Licence: Apache-2.0 license
A pickup line generator using Tensorflow

Programming Languages

python
139335 projects - #7 most used programming language
javascript
184084 projects - #8 most used programming language

Credit

This project was forked from https://github.com/martin-gorner/tensorflow-rnn-shakespeare

assume all code is attributed to and owned by Martin Gorner.

Original Code And Presentation

Code for the Recurrent Neural Network in the presentation "Tensorflow and deep learning - without a PhD, Part 2"

The presentation itself is available here:

This sample has now been updated for Tensorflow 1.1. Please make sure you redownload the checkpoint files if you use rnn_play.py.

Usage:

To Train

Step 1 - Train on Twitter Data

$ python3 rnn_train.py This will train the model and save it with checkpoint files.

After training, you can see what the model outputs by using

$ python3 rnn_play.py

You will need to replace the files with your checkpoint files.

Step 2 - Train on pickup line data

$ python3 rnn_retrain.py

You will need to replace the checkpoint file names for this to work.

To Play

Point to the correct checkpoint files and do $ python3 rnn_play.py

Each training session will generate saved checkpoints as well as save a final model when it has finished. you can monitor the progress using Tensorboard

$ tensorboard --logdir="your/log/path"

Note that the project description data, including the texts, logos, images, and/or trademarks, for each open source project belongs to its rightful owner. If you wish to add or remove any projects, please contact us at [email protected].