tfm.uplift.layers.heads.two_tower_logits_head.LinearLayeringConfig

Configuration for the linear layering method.

tfm.uplift.layers.heads.two_tower_logits_head.LinearLayeringConfig(
    kernel_initializer: str = 'glorot_uniform',
    kernel_regularizer: (str | None) = None
)

Attributes
kernel_initializer	kernel initializer for the learnable weight matrix.
kernel_regularizer	kernel regularizer for the learnable weight matrix.

Methods

__eq__

__eq__(
    other
)

Return self==value.

Class Variables
kernel_initializer	'glorot_uniform'
kernel_regularizer	None