1cda66554188c4504aa94d1bcffd4cef9ee8a997,autokeras/net_transformer.py,,transform,#Any#,79

Before Change

def transform(model):
    models = []
    layers = model.layers
    for index in range(len(layers) - 1):
        if isinstance(layers[index], tuple(WEIGHTED_LAYER_FUNC_LIST)):
            models.append(to_deeper_model(model, index))
            models.append(to_wider_model(model, index))
    return models

After Change

        if is_conv_layer(layer):
            models.append(Sequential(to_deeper_conv_model(conv_model, layer), dense_model))
            models.append(Sequential(to_wider_conv_model(conv_model, layer), dense_model))
    models.append(Sequential(to_skip_connection_model(conv_model), dense_model))

    return models

Instances