在训练过程中如何设置重量的变化范围(约束)。该平台是Keras或TensorFlow

时间:2019-11-05 18:44:33

标签: keras

在训练过程中如何设置重量的变化范围(约束)?该平台是Keras或TensorFlow

1 个答案:

答案 0 :(得分:1)

您可以在这里查看:https://keras.io/constraints/

我认为您正在寻找MinMaxNorm()

keras.constraints.MinMaxNorm(min_value=0.0, max_value=1.0, rate=1.0, axis=0)

在您的代码中,您可以这样做:

from keras.constraints import max_norm
model.add(Dense(64, kernel_constraint=min_max_norm(min_value=0.0, max_value=1.0, rate=1.0, axis=0)))

我建议使用from tensorflow.keras.constraints import max_norm,因为趋势是切换到TensorFlow内部的Keras而非普通Keras。