utils.py 文件源码

def prep_model_for_vis(model, out_layer_name='predictions'):
    layer_idx = find_layer_idx(model, out_layer_name)

    model.layers[layer_idx].activation = activations.linear
    model = apply_modifications(model)
    return model