Encoder Reference
All encoders inherit from the abstract Encoder
defined in seq2seq.encoders.encoder
and receive params
, mode
arguments at instantiation time. Available hyperparameters vary by encoder class.
UnidirectionalRNNEncoder
Name | Default | Description |
---|---|---|
rnn_cell.cell_class |
BasicLSTMCell |
The class of the rnn cell. Cell classes can be fully defined (e.g. tensorflow.contrib.rnn.BasicRNNCell ) or must be in tf.contrib.rnn or seq2seq.contrib.rnn_cell . |
rnn_cell.cell_params |
{"num_units": 128} |
A dictionary of parameters to pass to the cell class constructor. |
rnn_cell.dropout_input_keep_prob |
1.0 |
Apply dropout to the (non-recurrent) inputs of each RNN layer using this keep probability. A value of 1.0 disables dropout. |
rnn_cell.dropout_output_keep_prob |
1.0 |
Apply dropout to the (non-recurrent) outputs of each RNN layer using this keep probability. A value of 1.0 disables dropout. |
rnn_cell.num_layers |
1 |
Number of RNN layers. |
rnn_cell.residual_connections |
False |
If true, add residual connections between RNN layers in the encoder. |
BidirectionalRNNEncoder
Same as the UnidirectionalRNNEncoder
. The same cell is used for forward and backward RNNs.
StackBidirectionalRNNEncoder
Same as the UnidirectionalRNNEncoder
. The same cell is used for forward and backward RNNs.
PoolingEncoder
An encoder that pools over embeddings, as described in https://arxiv.org/abs/1611.02344. The encoder supports optional positions embeddings and a configurable pooling window.
Name | Default | Description |
---|---|---|
pooling_fn |
tensorflow.layers.average_pooling1d |
The 1-d pooling function to use, e.g. tensorflow.layers.average_pooling1d . |
pool_size |
5 |
The pooling window, passed as pool_size to the pooling function. |
strides |
1 |
The stride during pooling, passed as strides the pooling function. |
position_embeddings.enable |
True |
If true, add position embeddings to the inputs before pooling. |
position_embeddings.combiner_fn |
tensorflow.add |
Function used to combine the position embeddings with the inputs. For example, tensorflow.add . |
position_embeddings.num_positions |
100 |
Size of the position embedding matrix. This should be set to the maximum sequence length of the inputs. |
InceptionV3Encoder
This encoder is experimental. This encoder puts the image through an InceptionV3 network and uses the last hidden layer before the logits as the feature representation.
Name | Default | Description |
---|---|---|
resize_height |
299 |
Resize the image to this height before feeding it into the convolutional network. |
resize_width |
299 |
Resize the image to this width before feeding it into the convolutional network. |