ClassificationOutputLayer.ConfigΒΆ

Component: ClassificationOutputLayer

class ClassificationOutputLayer.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, SoftHardBCELoss.Config] = CrossEntropyLoss.Config()

Default JSON

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