BasePairwiseModel.ConfigΒΆ
Component: BasePairwiseModel
-
class
BasePairwiseModel.
Config
[source] 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
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,
"activation": "relu"
},
"output_layer": {
"ClassificationOutputLayer": {
"load_path": null,
"save_path": null,
"freeze": false,
"shared_module_key": null,
"loss": {
"CrossEntropyLoss": {}
},
"label_weights": null
}
},
"encode_relations": true
}