tfm.uplift.types.TwoTowerNetworkOutputs

Tensors computed by a BaseTwoTowerUpliftNetwork layer.

tfm.uplift.types.TwoTowerNetworkOutputs(
    shared_embedding, control_logits, treatment_logits
)

shared_embedding embedding computed by the backbone layer and used as a shared representation for the control and treatment towers.
control_logits logits for the control group. Its shape and dtype must be the same as the treatment logits.
treatment_logits logits for the treatment group. Its shape and dtype must be the same as the control logits.
dtype

shape

Child Classes

class Spec

Methods

__eq__

__eq__(
    other
)

Return self==value.

__ne__

__ne__(
    other
)

Return self!=value.