fd80a2cfcc8a04df863a57ba7592182db0db5e0e,autokeras/generator.py,DefaultClassifierGenerator,generate,#DefaultClassifierGenerator#,51

Before Change

        output_tensor = Activation("relu")(output_tensor)
        output_tensor = Dropout(constant.CONV_DROPOUT_RATE)(output_tensor)

        output_tensor = Flatten()(output_tensor)
        output_tensor = Dense(128, activation="relu")(output_tensor)
        output_tensor = Dropout(constant.DENSE_DROPOUT_RATE)(output_tensor)
        output_tensor = Dense(128, activation="relu")(output_tensor)
        output_tensor = Dropout(constant.DENSE_DROPOUT_RATE)(output_tensor)
        output_tensor = Dense(self.n_classes, activation="softmax")(output_tensor)
        return Model(inputs=input_tensor, outputs=output_tensor)

After Change

        Return the default classifier model that has been compiled.
        pool = self._get_pool_layer_func()
        conv = get_conv_layer_func(len(self._get_shape(3)))
        ave = get_ave_layer_func(len(self._get_shape(3)))

        input_tensor = Input(shape=self.input_shape)
        output_tensor = conv(32, kernel_size=self._get_shape(3), padding="same")(input_tensor)

Instances