Skip to content

EmbeddingCombiner

Encodes boxes using a combination of multiple encoders

Parameters

PARAMETER DESCRIPTION
pipeline

The pipeline object

TYPE: Pipeline DEFAULT: None

name

The name of the pipe

TYPE: str DEFAULT: 'embedding-combiner'

mode

The mode to use to combine the encoders:

  • sum: Sum the outputs of the encoders
  • cat: Concatenate the outputs of the encoders

TYPE: Literal['sum', 'cat'] DEFAULT: 'sum'

dropout_p

Dropout probability used on the output of the box and textual encoders

TYPE: float DEFAULT: 0.0

encoders

The encoders to use. The keys are the names of the encoders and the values are the encoders themselves.

TYPE: TrainablePipe[EmbeddingOutput] DEFAULT: {}