b80fb3bdcfeac64d91b7365be1196392e013dcb8,autokeras/utils.py,,extract_config,#Any#,95

Before Change

def extract_config(network):
    Return configuration of one model
    config = {"type": [], "config": []}
    for layer in network.layers:
        name = type(layer).__name__
        config["type"].append(name)
        layer_config = layer.get_config()
        important_attr = {}
        for attr in LAYER_ATTR[name]:
            important_attr[attr] = layer_config[attr]
        config["config"].append(important_attr)
    return config


def ensure_dir(directory):

After Change

def extract_config(network):
    Return configuration of one model
    return network.get_config()


def ensure_dir(directory):
    Create directory if it does not exist

Instances