Baseline model for Cassava Leaf Disease classification

baseline_VGG16

def build_baseline_vgg():
    input_shape = (160, 120, 3)
    baseline_model = VGG16(weights='imagenet', include_top=False, input_shape=input_shape)
    x = baseline_model.output
    x = Flatten()(x)
    x = Dense(512, activation='relu')(x)
    x = Dropout(rate=0.25)(x)

    x = Dense(256, activation='relu')(x)
    x = Dropout(rate=0.25)(x)

    x = Dense(128, activation='relu')(x)
    x = Dropout(rate=0.25)(x)

    predictions = Dense(5, activation='softmax')(x)
    model = Model(inputs=baseline_model.input, outputs=predictions)
    model.compile(optimizer=_opt,
                  loss=_loss,
                  metrics=_metrics)
    return model