减缓可训练权重更新速度的包装器
keras-succ-reg-wrapper的Python项目详细描述
keras连续正则化包装器
减缓可训练权重更新速度的包装器。
安装
pip install keras-succ-reg-wrapper
用法
importkerasfromkeras_succ_reg_wrapperimportSuccReginput_layer=keras.layers.Input(shape=(1,),name='Input')dense_layer=SuccReg(layer=keras.layers.Dense(units=1,name='Dense'),regularizer=keras.regularizers.L1L2(l2=1e-3),# Any regularizername='Output',)(input_layer)model=keras.models.Model(inputs=input_layer,outputs=dense_layer)model.compile(optimizer='adam',loss='mse')model.summary()