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: 11 months ago
JSON representation
An implementation of skip-thought vectors in Tensorflow
- Host: GitHub
- URL: https://github.com/persiyanov/skip-thought-tf
- Owner: persiyanov
- License: mit
- Created: 2016-09-28T10:59:45.000Z (over 9 years ago)
- Default Branch: master
- Last Pushed: 2023-03-24T21:54:36.000Z (over 3 years ago)
- Last Synced: 2025-04-23T20:42:45.394Z (about 1 year ago)
- Topics: deep-learning, deeplearning, embeddings, nlp, skip-thought-vectors, tensorflow, text-summarization
- Language: Python
- Homepage:
- Size: 130 KB
- Stars: 74
- Watchers: 1
- Forks: 31
- Open Issues: 5
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
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:
....
```