Skip to content

A "Char-RNN" unconventional machine learning software using tensorflow.

License

Notifications You must be signed in to change notification settings

BenTimor/Tensorflow-CharRNN

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

2 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

(Yet Another) Tensorflow ChatRNN

Yeah I know, there are like million versions of Tensorflow ChatRNN. This is my version.

Training:

python train.py --data TEXT_PATH --save CHECKPOINTS_FOLDER

Evalling:

python eval.py --path CHECKPOINTS_FOLDER --prime PRIME

or

python eval.py --path CHECKPOINTS_FOLDER --loop 1

For example - Shakespeare

Yay, another machine learning which comes with tiny Shakespeare dataset.

Train it (It's already half-trained so you don't really have to do this step):

python train.py --data shakespeare/input.txt --save shakespeare/checkpoints/

Use it:

python eval.py --path shakespeare/checkpoints/ --loop 1