tf.contrib.learn.TensorFlowRNNClassifier.__init__()

tf.contrib.learn.TensorFlowRNNClassifier.__init__(rnn_size, n_classes, cell_type='gru', num_layers=1, input_op_fn=null_input_op_fn, initial_state=None, bidirectional=False, sequence_length=None, attn_length=None, attn_size=None, attn_vec_size=None, batch_size=32, steps=50, optimizer='Adagrad', learning_rate=0.1, class_weight=None, clip_gradients=5.0, continue_training=False, config=None, verbose=1)

Initializes a TensorFlowRNNClassifier instance.

Args:
  • rnn_size: The size for rnn cell, e.g. size of your word embeddings.
  • cell_type: The type of rnn cell, including rnn, gru, and lstm.
  • num_layers: The number of layers of the rnn model.
  • input_op_fn: Function that will transform the input tensor, such as creating word embeddings, byte list, etc. This takes an argument x for input and returns transformed x.
  • bidirectional: boolean, Whether this is a bidirectional rnn.
  • sequence_length: If sequence_length is provided, dynamic calculation is performed. This saves computational time when unrolling past max sequence length.
  • initial_state: An initial state for the RNN. This must be a tensor of appropriate type and shape [batch_size x cell.state_size].
  • attn_length: integer, the size of attention vector attached to rnn cells.
  • attn_size: integer, the size of an attention window attached to rnn cells.
  • attn_vec_size: integer, the number of convolutional features calculated on attention state and the size of the hidden layer built from base cell state.
  • n_classes: Number of classes in the target.
  • batch_size: Mini batch size.
  • steps: Number of steps to run over data.
  • optimizer: Optimizer name (or class), for example "SGD", "Adam", "Adagrad".
  • learning_rate: If this is constant float value, no decay function is used. Instead, a customized decay function can be passed that accepts global_step as parameter and returns a Tensor. e.g. exponential decay function:

    def exp_decay(global_step):
        return tf.train.exponential_decay(
            learning_rate=0.1, global_step,
            decay_steps=2, decay_rate=0.001)
    
  • class_weight: None or list of n_classes floats. Weight associated with classes for loss computation. If not given, all classes are supposed to have weight one.

  • continue_training: when continue_training is True, once initialized model will be continuely trained on every call of fit.

  • config: RunConfig object that controls the configurations of the session, e.g. num_cores, gpu_memory_fraction, etc.

doc_TensorFlow
2016-10-14 13:07:03
Comments
Leave a Comment

Please login to continue.