DenseRetrievalOutputLayer.Config

Component: DenseRetrievalOutputLayer

class DenseRetrievalOutputLayer.Config

Bases: PairwiseCosineDistanceOutputLayer.Config

All Attributes (including base classes)

load_path: Optional[str] = None

save_path: Optional[str] = None

freeze: bool = False

shared_module_key: Optional[str] = None

loss: Union[BinaryCrossEntropyLoss.Config, CosineEmbeddingLoss.Config, MAELoss.Config, MSELoss.Config, NLLLoss.Config] = CosineEmbeddingLoss.Config()

score_threshold: float = 0.9

score_type: OutputScore = <OutputScore.norm_cosine: 2>

label_weights: Optional[dict[str, float]] = None

Default JSON

{
    "load_path": null,
    "save_path": null,
    "freeze": false,
    "shared_module_key": null,
    "loss": {
        "CosineEmbeddingLoss": {
            "margin": 0.0
        }
    },
    "score_threshold": 0.9,
    "score_type": 2,
    "label_weights": null
}