merlin.models.tf.CrossBlock

merlin.models.tf.CrossBlock(depth: int = 1, filter: Optional[Union[merlin.schema.schema.Schema, merlin.schema.tags.Tags, List[str], merlin.models.tf.core.tabular.Filter]] = None, low_rank_dim: Optional[int] = None, use_bias: bool = True, kernel_initializer: Union[str, keras.initializers.initializers_v2.Initializer] = 'truncated_normal', bias_initializer: Union[str, keras.initializers.initializers_v2.Initializer] = 'zeros', kernel_regularizer: Optional[Union[str, keras.regularizers.Regularizer]] = None, bias_regularizer: Optional[Union[str, keras.regularizers.Regularizer]] = None, inputs: Optional[keras.engine.base_layer.Layer] = None, **kwargs) → merlin.models.tf.core.combinators.SequentialBlock[source]
This block provides a way to create high-order feature interactions

by a number of stacked Cross Layers, from DCN V2: Improved Deep & Cross Network [1]. See Eq. (1) for full-rank and Eq. (2) for low-rank version.

References

Parameters

  • depth (int, optional) – Number of cross-layers to be stacked, by default 1

  • filter (Optional[Union[Schema, Tags, List[str], Filter]], optional) – Features filter to be applied on the input, by default None

  • low_rank_dim (Optional[int], optional) – If this argument is provided, the weight (W in R(dxd)), where d is the input features dimension matrix, is factorized in a low-rank matrix W = U*V where U and D have (dxr) shape and low_rank_dim = r, by default None

  • use_bias (bool, optional) – Enables or not the bias term, by default True

  • kernel_initializer (InitializerType, optional) – Initializer to use on the kernel matrix, by default “truncated_normal”

  • bias_initializer (InitializerType, optional) – Initializer to use on the bias vector, by default “zeros”

  • kernel_regularizer (Optional[RegularizerType], optional) – Regularizer to use on the kernel matrix, by default None

  • bias_regularizer (Optional[RegularizerType], optional) – Regularizer to use on the bias vector, by default None

  • inputs (Optional[tf.keras.layers.Layer], optional) – If an InputBlock is provided, this block checks if features are being aggregated with concat, otherwise it does that, as cross blocks need features to be aggregated before, by default None

Returns

A SequentialBlock with a number of stacked Cross layers

Return type

SequentialBlock

Raises

ValueError – Number of cross layers (depth) should be positive