models.py 文件源码

def get_maxout(
        size, loss='categorical_crossentropy', optimizer=Adam,
        optimizer_kwargs={}):
    # MaxOut network
    model = Sequential()
    model.add(MaxoutDense(256, input_shape=(size,), nb_feature=5,
                          init='he_uniform'))
    model.add(MaxoutDense(128, nb_feature=5))
    model.add(Dense(64))
    model.add(Activation('relu'))
    model.add(Dense(25))
    model.add(Activation('relu'))
    model.add(Dense(2))
    model.add(Activation('sigmoid'))
    optimizer = optimizer(**optimizer_kwargs)
    model.compile(loss=loss, optimizer=optimizer)
    return model