Module: tfm.uplift.types

Defines types used by the keras uplift modeling library.

Classes

class TwoTowerNetworkOutputs: Tensors computed by a BaseTwoTowerUpliftNetwork layer.

class TwoTowerPredictionOutputs: Inference tensors computed by a TwoTowerUpliftNetwork layer.

class TwoTowerTrainingOutputs: Training tensors computed by a TwoTowerUpliftNetwork layer.

Type Aliases

DictOfTensors

ListOfTensors

TupleOfTensors

CollectionOfTensors Instance of types.UnionType
TensorType Instance of types.UnionType