import tensorflow as tf

    def id2tag(self, pred_ids, name=None):
        mapping_strings = self.load_tag_data()
        reverse_vocab_tags = tf.contrib.lookup.index_to_string_table_from_tensor(
            mapping_strings, name=name
        )

        pred_strings = reverse_vocab_tags.lookup(tf.to_int64(pred_ids))

        return pred_strings

    def id2word(self, word_ids, name=None):
        mapping_strings = self.load_word_data()
