import tensorflow as tf


        # word_strings = self.id2word(word_ids, name='word_strings')

        # print(word_strings)

        if self.mode == tf.estimator.ModeKeys.PREDICT:

            predictions = {"pred_ids": pred_ids, "tags": pred_strings}
            return tf.estimator.EstimatorSpec(self.mode, predictions=predictions)
        else:
            # true_tag_ids = self.labels
            true_tag_ids = self.tag2id(self.labels, "labels")

            # print(pred_strings)
            # print(self.labels)

