merlin.models.tf.OptimizerBlocks

class merlin.models.tf.OptimizerBlocks(optimizer: Union[str, keras.optimizers.optimizer.Optimizer], blocks: Sequence[merlin.models.tf.core.base.Block])

Bases: object

dataclass for a pair of optimizer and blocks that the optimizer should apply to. .. rubric:: Example

ml.OptimizerBlocks(“sgd”, [user_tower, third_tower]) ml.OptimizerBlocks(“adam”, item_tower)

__init__(optimizer: Union[str, keras.optimizers.optimizer.Optimizer], blocks: Sequence[merlin.models.tf.core.base.Block]) → None

Methods

__init__(optimizer, blocks)
from_config(config)
get_config()	return a tuple of serialized keras objects

Attributes

optimizer
blocks

optimizer: Union[str, keras.optimizers.optimizer.Optimizer]

blocks: Sequence[merlin.models.tf.core.base.Block]

get_config()

return a tuple of serialized keras objects

classmethod from_config(config)