13aa31bc1a0b87d6c4b5e787c9b041ec83831c34,autokeras/blocks/reduction.py,SpatialReduction,build,#SpatialReduction#Any#Any#,106

Before Change

                                                           "global_max",
                                                           "global_avg"],
                                                          default="global_avg")
        if reduction_type == "flatten":
            output_node = Flatten().build(hp, output_node)
        elif reduction_type == "global_max":
            output_node = layer_utils.get_global_max_pooling(
                output_node.shape)()(output_node)
        elif reduction_type == "global_avg":
            output_node = layer_utils.get_global_average_pooling(
                output_node.shape)()(output_node)
        return output_node


class TemporalReduction(block_module.Block):

After Change

        if len(output_node.shape) <= 2:
            return output_node

        if self.reduction_type is None:
            reduction_type = hp.Choice(
                REDUCTION_TYPE, [FLATTEN, GLOBAL_MAX, GLOBAL_AVG])
            with hp.conditional_scope(REDUCTION_TYPE, [reduction_type]):
                return self._build_block(hp, output_node, reduction_type)
        else:
            return self._build_block(hp, output_node, self.reduction_type)

    def _build_block(self, hp, output_node, reduction_type):
        if reduction_type == FLATTEN:
            output_node = Flatten().build(hp, output_node)
        elif reduction_type == GLOBAL_MAX:

Instances