transformers4rec.tf.tabular package

Submodules

transformers4rec.tf.tabular.aggregation module

class transformers4rec.tf.tabular.aggregation.ConcatFeatures(*args, **kwargs)

Bases: transformers4rec.config.schema.SchemaMixin, keras.engine.base_layer.Layer, merlin_standard_lib.registry.RegistryMixin[TabularAggregation], abc.ABC

call(inputs: Dict[str, tensorflow.python.framework.ops.Tensor], **kwargs) → tensorflow.python.framework.ops.Tensor

compute_output_shape(input_shapes)

get_config()

class transformers4rec.tf.tabular.aggregation.StackFeatures(*args, **kwargs)

Bases: transformers4rec.config.schema.SchemaMixin, keras.engine.base_layer.Layer, merlin_standard_lib.registry.RegistryMixin[TabularAggregation], abc.ABC

call(inputs: Dict[str, tensorflow.python.framework.ops.Tensor], **kwargs) → tensorflow.python.framework.ops.Tensor

compute_output_shape(input_shapes)

get_config()

class transformers4rec.tf.tabular.aggregation.ElementwiseFeatureAggregation(*args, **kwargs)

Bases: transformers4rec.config.schema.SchemaMixin, keras.engine.base_layer.Layer, merlin_standard_lib.registry.RegistryMixin[TabularAggregation], abc.ABC

class transformers4rec.tf.tabular.aggregation.ElementwiseSum(*args, **kwargs)

Bases: transformers4rec.config.schema.SchemaMixin, keras.engine.base_layer.Layer, merlin_standard_lib.registry.RegistryMixin[TabularAggregation], abc.ABC

call(inputs: Dict[str, tensorflow.python.framework.ops.Tensor], **kwargs) → tensorflow.python.framework.ops.Tensor

compute_output_shape(input_shape)

class transformers4rec.tf.tabular.aggregation.ElementwiseSumItemMulti(*args, **kwargs)

Bases: transformers4rec.config.schema.SchemaMixin, keras.engine.base_layer.Layer, merlin_standard_lib.registry.RegistryMixin[TabularAggregation], abc.ABC

call(inputs: Dict[str, tensorflow.python.framework.ops.Tensor], **kwargs) → tensorflow.python.framework.ops.Tensor

compute_output_shape(input_shape)

get_config()

REQUIRES_SCHEMA = True

transformers4rec.tf.tabular.tabular module

transformers4rec.tf.tabular.transformations module

class transformers4rec.tf.tabular.transformations.AsSparseFeatures(*args, **kwargs)

Bases: transformers4rec.config.schema.SchemaMixin, keras.engine.base_layer.Layer, merlin_standard_lib.registry.RegistryMixin[TabularTransformation], abc.ABC

call(inputs: Dict[str, tensorflow.python.framework.ops.Tensor], **kwargs) → Dict[str, tensorflow.python.framework.ops.Tensor]

compute_output_shape(input_shape)

class transformers4rec.tf.tabular.transformations.AsDenseFeatures(*args, **kwargs)

Bases: transformers4rec.config.schema.SchemaMixin, keras.engine.base_layer.Layer, merlin_standard_lib.registry.RegistryMixin[TabularTransformation], abc.ABC

call(inputs: Dict[str, tensorflow.python.framework.ops.Tensor], **kwargs) → Dict[str, tensorflow.python.framework.ops.Tensor]

compute_output_shape(input_shape)

class transformers4rec.tf.tabular.transformations.StochasticSwapNoise(*args, **kwargs)

Bases: transformers4rec.config.schema.SchemaMixin, keras.engine.base_layer.Layer, merlin_standard_lib.registry.RegistryMixin[TabularTransformation], abc.ABC

Applies Stochastic replacement of sequence features

call(inputs: Union[tensorflow.python.framework.ops.Tensor, Dict[str, tensorflow.python.framework.ops.Tensor]], input_mask: Optional[tensorflow.python.framework.ops.Tensor] = None, training=True, **kwargs) → Union[tensorflow.python.framework.ops.Tensor, Dict[str, tensorflow.python.framework.ops.Tensor]]

augment(input_tensor: tensorflow.python.framework.ops.Tensor, mask: Optional[tensorflow.python.framework.ops.Tensor], **kwargs) → tensorflow.python.framework.ops.Tensor

compute_output_shape(input_shape)

get_config()

Module contents