9a387157f03916b2f4f9f63647792dca39942471,autokeras/generator.py,DefaultClassifierGenerator,generate,#DefaultClassifierGenerator#,50

Before Change

        output_tensor = Dense(128, activation="relu")(output_tensor)
        output_tensor = Dense(self.n_classes, activation="softmax")(output_tensor)

        model = Model(input_tensor, output_tensor)
        model.compile(loss=categorical_crossentropy,
                      optimizer=Adadelta(),
                      metrics=["accuracy"])
        return model


class RandomConvClassifierGenerator(ClassifierGenerator):
    A classifier generator that generates random convolutional neural networks.

After Change

        output_tensor = pool(padding="same")(output_tensor)

        output_tensor = conv(64, kernel_size=self._get_shape(3), padding="same", activation="linear")(output_tensor)
        output_tensor = BatchNormalization()(output_tensor)
        output_tensor = Activation("relu")(output_tensor)

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


class RandomConvClassifierGenerator(ClassifierGenerator):
    A classifier generator that generates random convolutional neural networks.

Instances