Ecosyste.ms: Awesome

An open API service indexing awesome lists of open source software.

https://github.com/persiyanov/skip-thought-tf

An implementation of skip-thought vectors in Tensorflow
https://github.com/persiyanov/skip-thought-tf

deep-learning deeplearning embeddings nlp skip-thought-vectors tensorflow text-summarization

Last synced: 8 days ago
JSON representation

An implementation of skip-thought vectors in Tensorflow

Lists

README

        

# skip-thought-tf
An implementation of skip-thought vectors in Tensorflow

# Usage
```python
from skipthought import SkipthoughtModel
from skipthought.data_utils import TextData
from skipthought.utils import seq2seq_triples_data_iterator

model = SkipthoughtModel(...)

td = TextData("path/to/data")
lines = td.dataset

prev, curr, next = td.make_triples(lines)
it = td.triples_data_iterator(prev, curr, next, td.max_len, batch_size)

with tf.Session() as sess:
init = tf.initialize_all_variables()
sess.run(init)

for enc_inp, prev_inp, prev_targ, next_inp, next_targ in it:
....

```