ConvLSTMCell для классификации текста - PullRequest
Новая
       26

ConvLSTMCell для классификации текста

0 голосов
/ 09 декабря 2018

Я пытаюсь использовать tf.contrib.rnn.ConvLSTMCell для классификации текста, но я получил следующее значение ошибки при его использовании: 

ValueError: Conv Linear expects 3D, 4D or 5D arguments: [[None, 303], [None, 86, 1]]

мой код: 

    with tf.name_scope('dynamic_rnn'):
        lstm_cell_fw = tf.contrib.rnn.ConvLSTMCell(
            conv_ndims= 1,
            input_shape= [self.max_sentence_len, self.embedding_dim],
            output_channels=1,
            kernel_shape=[1, 1],
            use_bias=True,
            skip_connection=False,
            forget_bias=1.0,
            initializers=None,
            name='conv_lstm_cell'

        )
        lstm_cell_bw = tf.contrib.rnn.ConvLSTMCell(
            conv_ndims= 1,
            input_shape= [self.max_sentence_len, self.embedding_dim],
            output_channels=1,
            kernel_shape=[1, 1],
            use_bias=True,
            skip_connection=False,
            forget_bias=1.0,
            initializers=None,
            name='conv_lstm_cell'

        )
        outputs, state, _ = tf.nn.static_bidirectional_rnn(
            lstm_cell_fw,
            lstm_cell_bw,
            tf.unstack(tf.transpose(inputs, perm=[1, 0, 2])),
            sequence_length=self.sentence_lens,
            dtype=tf.float32,
            scope=None
        )
        outputs = tf.reshape(tf.concat(outputs, 1), [-1, self.max_sentence_len, self.n_hidden * 2])
        batch_size = tf.shape(outputs)[0]

Пожалуйста, кто-нибудь может мне помочь?Я новичок в этой области

...