61613b27b92aef20d87c881cfc7451e2b942df00,python/mead/tf/exporters.py,TaggerTensorFlowExporter,_create_model,#TaggerTensorFlowExporter#Any#Any#,151

Before Change

        model = load_tagger_model(basename, sess=sess, **kwargs)
        softmax_output = tf.nn.softmax(model.probs)
        values, indices = tf.nn.top_k(softmax_output, 1)
        return model, model.best, values

    def _create_rpc_call(self, sess, basename):
        model, classes, values = self._create_model(sess, basename)

After Change

                list_of_labels[idval] = label
            class_tensor = tf.constant(list_of_labels)
            table = tf.contrib.lookup.index_to_string_table_from_tensor(class_tensor)
            classes = table.lookup(tf.to_int64(indices))
            return model, classes, values
        else:
            return model, indices, values

Instances