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

Before Change

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

        reduction_type = self.reduction_type or hp.Choice("reduction_type",
                                                          ["flatten",
                                                           "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(

After Change

            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):

Instances