skip-thought-tf
An implementation of skip-thought vectors in Tensorflow
Usage
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:
....