add_weights_regularizer

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

將權重正則化器加入提供的 Tensor

參數

  • variable: Variable。要加上正則化的 Tensor。
  • loss: str。正則化模式。
  • weight_decay: float。用於正則化的衰退。
  • add_to_collection: str。將正則化損失加入指定的集合。預設為：tf.GraphKeys.REGULARIZATION_LOSSES。

回傳值

tf.Tensor。權重正則化器。