Tensorflow：TypeError：helper必须是Helper，收到：

Hello I am trying to create a BasicDecoder with a GreedyEmbeddingHelper but it is giving an error:

您好我正在尝试使用GreedyEmbeddingHelper创建BasicDecoder，但它给出了一个错误：

TypeError: helper must be a Helper, received: <class 'helper.GreedyEmbeddingHelper'>

Here is a simplified version of my code:

这是我的代码的简化版本：

        elif self.mode == 'decode':

            # Start_tokens: [batch_size,] `int32` vector
            start_tokens = tf.ones([self.batch_size, self.dimension], tf.float32) * 0.1337
            end_token = 0.1337

            def project_inputs(inputs):
                print inputs.shape
                return input_layer(inputs)

            if not self.use_beamsearch_decode:
                # Helper to feed inputs for greedy decoding: uses the argmax of the output
                decoding_helper = helper.GreedyEmbeddingHelper(start_tokens=start_tokens,
                                                                end_token=end_token,
                                                                embedding=project_inputs)
                # Basic decoder performs greedy decoding at each time step
                print("building greedy decoder..")
                inference_decoder = seq2seq.BasicDecoder(cell=self.decoder_cell,
                                                         helper=decoding_helper,
                                                         initial_state=self.decoder_initial_state,
                                                         output_layer=output_layer)
            else:
                # Beamsearch is used to approximately find the most likely translation
                print("building beamsearch decoder..")
                inference_decoder = beam_search_decoder.BeamSearchDecoder(cell=self.decoder_cell,
                                                           embedding=project_inputs,
                                                           start_tokens=start_tokens,
                                                           end_token=end_token,
                                                           initial_state=self.decoder_initial_state,
                                                           beam_width=self.beam_width,
                                                           output_layer=output_layer,)

I don't know how to fix it because Helper is an abstract class. So it won't be possible.

我不知道如何修复它，因为Helper是一个抽象类。所以这是不可能的。

1 个解决方案

#1

GreedyEmbeddingHelper is defined at tf.contrib.seq2seq.GreedyEmbeddingHelper. So instead of helper.GreedyEmbeddingHelper, use tf.contrib.seq2seq.GreedyEmbeddingHelper

GreedyEmbeddingHelper在tf.contrib.seq2seq.GreedyEmbeddingHelper中定义。因此，而不是helper.GreedyEmbeddingHelper，使用tf.contrib.seq2seq.GreedyEmbeddingHelper