Kensuke-Mitsuzawa / tweet2vec

Twitter hashtag prediction

Tweet2Vec

This repository provides a character-level encoder/trainer for social media posts. See Tweet2Vec paper for details.

This code is available as python package.

Prerequisites

• Python
  • Python2.7
  • Python3.5
• Theano and all dependencies (latest)
• Lasagne (latest)
• Numpy
• Maybe more, just use pip install if you get an error

Setup

Install

I recommend to make separated python environment with virtualenv or conda etc.

python setup.py install

Hints

You might see error messages during installing following packages

• numpy
• scipy
• lasagne
• Theano

In that case, you're supposed to make install manually.

If you're using conda under anaconda-distribution, it's easy to install them.

You install packages with conda

$ conda install numpy scipy theano

Conda does not have lasagne, so you install it with pip

$ pip install lasagne

Data and Preprocessing

Unfortunately we are not allowed to release the data used in experiments from the paper, due to licensing restrictions. Hence, we describe the data format and preprocessing here -

1. Preprocessing - We replace HTML tags, usernames, and URLs from tweet text with special tokens. Hashtags are also removed from the body of a tweet, and re-tweets are discarded. Example code is provided in misc/preprocess.py.

2. Encoding File Format - If you have a bunch of posts that you want to embed into a vector space, use the _encoder.sh scripts provided. The input file must contain one tweet per line (make sure you preprocess these first). An example is provided in misc/encoder_example.txt.

3. Training File Format - To train the models from scratch, use the _trainer.sh scripts provided. The input file must contain one (hashtag,tweet) pair per line separated by a tab. There should be only one tag per line - for tweets with multiple tags split them into separate line. See misc/trainer_example.txt for an example.

4. Test/Validation File Format - After training the model, you can test it on a held-out set using _tester.sh scripts provided. It has the same format as the training file format, except it can have multiple tags per separated by a comma. Example in misc/tester_example.txt.

Example code

You can check example code which shows you how to use this package.

See example.py for detail.

Note

Make sure to add THEANO_FLAGS=device=cpu,floatX=float32 before any command if you are running on a CPU.

like

THEANO_FLAGS=device=cpu,floatX=float32 python example.py

Contributors

Bhuwan Dhingra, Dylan Fitzpatrick, Zhong Zhou, Michael Muehl. Special thanks to Yun Fu for the preprocessing JAR-file.

If you end up using this code, please cite the following paper -

Dhingra, Bhuwan, Zhong Zhou, Dylan Fitzpatrick, Michael Muehl, and William W. Cohen. "Tweet2Vec: Character-Based Distributed Representations for Social Media." ACL (2016).

@article{dhingra2016tweet2vec,
  title={Tweet2Vec: Character-Based Distributed Representations for Social Media},
  author={Dhingra, Bhuwan and Zhou, Zhong and Fitzpatrick, Dylan and Muehl, Michael and Cohen, William W},
  journal={ACL},
  year={2016}
}

Report bugs and missing info to bdhingraATandrewDOTcmuDOTedu (replace AT, DOT appropriately).