074d16ac217b9cddc04d093d019c61e393522eb5,autokeras/net_transformer.py,,to_wider_dense_model,#Any#Any#,73

Before Change

def to_wider_dense_model(model, level):
    next_wider_layer, ind = get_next_dense_conv(level, model.layers)
    n_add = randint(1, 4 * model.layers[level].units)
    new_wider_layer, new_next_wider_layer = dense_to_wider_layer(model.layers[level], next_wider_layer, n_add)
    return replace_layers(model, [level, ind], [new_wider_layer, new_next_wider_layer])

After Change

def to_wider_dense_model(model, target):
    graph = Graph(model)
    n_add = randint(1, 4 * target.units)
    return graph.to_dense_wider_model(target, n_add)


def to_deeper_conv_model(model, target):
    graph = Graph(model)

Instances