Tweet like Trump

Train an LSTM model to generate Trump style tweets.

Dependencies

• Python 3.5+
• PyTorch 0.4.0

Usage

1. Clone this repo

$ git clone https://github.com/rliu054/tweet-like-trump
$ cd tweet-like-trump/

2. Run the model

$ python main.py --embed_size=50 --hidden_size=50 --epochs=20

List of parameters for the model:

usage: main.py [-h] [--data DATA] [--batch_size BATCH_SIZE] [--epochs EPOCHS]
               [--embed_size EMBED_SIZE] [--hidden_size HIDDEN_SIZE]
               [--num_layers NUM_LAYERS] [--seq_len SEQ_LEN]
               [--dropout DROPOUT] [--clips CLIPS] [--lr LR]

Generate tweets in Trump style

optional arguments:
  -h, --help            show this help message and exit
  --data DATA           data corpus location
  --batch_size BATCH_SIZE
                        batch size
  --epochs EPOCHS       number of epochs to run
  --embed_size EMBED_SIZE
                        embedding size
  --hidden_size HIDDEN_SIZE
                        hidden layer size
  --num_layers NUM_LAYERS
                        layers of RNN to be stacked
  --seq_len SEQ_LEN     sequence length
  --dropout DROPOUT     dropout ratio
  --clips CLIPS         used to clip gradients
  --lr LR               learning rate

3. Check the generate tweets!

$ cat output.txt

Sample output

Here are some samples from the generates tweets. We can see some iconic phrases the POTUS love to use.

* please mr trump please run for potus thank you <EOS> 
* thank you virginia and i will be on on monday at 900 pm <EOS> 
* im of paul ryan is a gorgeous guy <EOS> 
* you should be president you would make my vote right save america <EOS> 
* if obamacare doesnt have the answers <EOS> 
* keep calling my big plan for the american worker ratings right because i had the course right now also true <EOS> 

Acknowledgements

The data used to train this model was grabbed from this great project.

Author

© Rui Liu