41 lines
1.2 KiB
Python
41 lines
1.2 KiB
Python
|
from keras_core import regularizers
|
||
|
from keras_core.api_export import keras_core_export
|
||
|
from keras_core.layers.layer import Layer
|
||
|
|
||
|
|
||
|
@keras_core_export("keras_core.layers.ActivityRegularization")
|
||
|
class ActivityRegularization(Layer):
|
||
|
"""Layer that applies an update to the cost function based input activity.
|
||
|
|
||
|
Args:
|
||
|
l1: L1 regularization factor (positive float).
|
||
|
l2: L2 regularization factor (positive float).
|
||
|
|
||
|
Input shape:
|
||
|
Arbitrary. Use the keyword argument `input_shape`
|
||
|
(tuple of integers, does not include the samples axis)
|
||
|
when using this layer as the first layer in a model.
|
||
|
|
||
|
Output shape:
|
||
|
Same shape as input.
|
||
|
"""
|
||
|
|
||
|
def __init__(self, l1=0.0, l2=0.0, **kwargs):
|
||
|
super().__init__(
|
||
|
activity_regularizer=regularizers.L1L2(l1=l1, l2=l2), **kwargs
|
||
|
)
|
||
|
self.supports_masking = True
|
||
|
self.l1 = l1
|
||
|
self.l2 = l2
|
||
|
|
||
|
def call(self, inputs):
|
||
|
return inputs
|
||
|
|
||
|
def compute_output_shape(self, input_shape):
|
||
|
return input_shape
|
||
|
|
||
|
def get_config(self):
|
||
|
base_config = super().get_config()
|
||
|
config = {"l1": self.l1, "l2": self.l2}
|
||
|
return {**base_config, **config}
|