BasePairwiseModel.Config

Component: BasePairwiseModel

class BasePairwiseModel.Config

Bases: BaseModel.Config

All Attributes (including base classes)

inputs: ModelInput = ModelInput()

decoder: MLPDecoder.Config = MLPDecoder.Config()

output_layer: Union[ClassificationOutputLayer.Config, PairwiseCosineDistanceOutputLayer.Config] = ClassificationOutputLayer.Config()

encode_relations: bool = True

Subclasses

• BertPairwiseModel.Config
• _EncoderPairwiseModel.Config
• BertPairwiseRegressionModel.Config
• PairwiseModel.Config
• QueryDocPairwiseRankingModel.Config

Default JSON

{
    "inputs": {},
    "decoder": {
        "load_path": null,
        "save_path": null,
        "freeze": false,
        "shared_module_key": null,
        "hidden_dims": [],
        "out_dim": null,
        "layer_norm": false,
        "dropout": 0.0,
        "bias": true,
        "activation": "relu",
        "temperature": 1.0,
        "spectral_normalization": false
    },
    "output_layer": {
        "ClassificationOutputLayer": {
            "load_path": null,
            "save_path": null,
            "freeze": false,
            "shared_module_key": null,
            "loss": {
                "CrossEntropyLoss": {}
            },
            "label_weights": null
        }
    },
    "encode_relations": true
}