add_weights_regularizer

tflearn.helpers.regularizer.add_weights_regularizer (variable, loss='L2', weight_decay=0.001, add_to_collection=None)

为提供的张量添加权重正则化器

参数

  • variable: Variable. 要添加正则化的张量。
  • loss: str. 正则化模式。
  • weight_decay: float. 用于正则化的衰减值。
  • add_to_collection: str. 将正则化损失添加到指定的集合中。默认值：tf.GraphKeys.REGULARIZATION_LOSSES。

返回值

tf.Tensor. 权重正则化器。