tfm.uplift.types.TwoTowerPredictionOutputs

Inference tensors computed by a TwoTowerUpliftNetwork layer.

Inherits From: TwoTowerNetworkOutputs

tfm.uplift.types.TwoTowerPredictionOutputs(
    shared_embedding,
    control_logits,
    treatment_logits,
    control_predictions,
    treatment_predictions,
    uplift
)

Attributes
control_predictions	predictions for the control group. Its shape and dtype must be the same as the treatment predictions.
treatment_predictions	predictions for the treatment group. Its shape and dtype must be the same as the control predictions.
uplift	difference between the treatment and control predictions.
dtype
shape

Child Classes

class Spec

Methods

__eq__

__eq__(
    other
)

Return self==value.

__ne__

__ne__(
    other
)

Return self!=value.