在训练之前使用一些权重实例化Keras模型

时间:2020-08-30 08:09:28

标签: keras

我有Keras模型:预先训练的简历模型+顶部增加了几层

我希望能够在model.fit之前进行model.predict

问:如何从屏幕截图中实例化一些权重(随机,零或其他)的模型enter image description here

1 个答案:

答案 0 :(得分:1)

这是一个虚拟示例,用于使用一些权重(随机,零或其他任何值)初始化模型

def base_model(xx):
    
    x = Dense(32)(xx)
    x = Dense(8)(x)
    
    return Model(xx,x)

inp = Input((32,32,3))
x = base_model(inp)
x = GlobalAveragePooling2D()(x.output)
x = Dropout(0.3)(x)
out = Dense(10, activation='softmax')(x)

model = Model(inp,out)
model.summary()

# set weight with random number from a uniform... you can do the same also with zeros...
model.set_weights([np.random.uniform(0,1, i.shape) for i in model.get_weights()])