seq2seq Namespace Reference

Module caffe2.python.examples.seq2seq. More...

Classes
class	Seq2SeqModelCaffe2

Functions
def	prepare_batch (batch)
def	gen_vocab (corpus, unk_threshold)
def	get_numberized_sentence (sentence, vocab)
def	gen_batches (source_corpus, target_corpus, source_vocab, target_vocab, batch_size, max_length)
def	run_seq2seq_model (args, model_params=None)
def	run_seq2seq_rnn_unidirection_with_no_attention (args)
def	main ()

Variables
	logger = logging.getLogger(__name__)
	Batch
string	EOS = '<EOS>'
string	UNK = '<UNK>'
string	GO = '<GO>'
string	PAD = '<PAD>'

Detailed Description

Module caffe2.python.examples.seq2seq.

Variable Documentation

Batch

seq2seq.Batch

Initial value:

=  collections.namedtuple('Batch', [
    'encoder_inputs',
    'encoder_lengths',
    'decoder_inputs',
    'decoder_lengths',
    'targets',
    'target_weights',
])

Definition at line 27 of file seq2seq.py.