4b2a2f18a4da11d9573d341c38db2c754c77fff5,autokeras/auto_model.py,AutoModel,_postprocess,#AutoModel#Any#,190

Before Change

    def _postprocess(self, y):
        y = nest.flatten(y)
        if not self._label_encoders:
            return y
        new_y = []
        for temp_y, label_encoder in zip(y, self._label_encoders):
            if label_encoder:
                new_y.append(label_encoder.decode(temp_y))

After Change

        y = nest.flatten(y)
        new_y = []
        for temp_y, head_block in zip(y, self.heads):
            if isinstance(head_block, head.Head):
                temp_y = head_block.postprocess(temp_y)
            new_y.append(temp_y)
        return new_y

Instances