WordTaggingOutputLayer.ConfigΒΆ

Component: WordTaggingOutputLayer

class WordTaggingOutputLayer.Config[source]

Bases: OutputLayerBase.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[CrossEntropyLoss.Config, BinaryCrossEntropyLoss.Config, AUCPRHingeLoss.Config, KLDivergenceBCELoss.Config, KLDivergenceCELoss.Config, LabelSmoothedCrossEntropyLoss.Config, StructuredMarginLoss.Config] = CrossEntropyLoss.Config()
label_weights: dict[str, float] = {}
ignore_pad_in_loss: Optional[bool] = True

Default JSON

{
    "load_path": null,
    "save_path": null,
    "freeze": false,
    "shared_module_key": null,
    "loss": {
        "CrossEntropyLoss": {}
    },
    "label_weights": {},
    "ignore_pad_in_loss": true
}