MulticlassOutputLayer.ConfigΒΆ
Component: MulticlassOutputLayer
-
class
MulticlassOutputLayer.
Config
Bases:
ClassificationOutputLayer.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, BinaryCrossEntropyWithLogitsLoss.Config, MultiLabelSoftMarginLoss.Config, AUCPRHingeLoss.Config, HingeLoss.Config, KLDivergenceBCELoss.Config, KLDivergenceCELoss.Config, LabelSmoothedCrossEntropyLoss.Config] = CrossEntropyLoss.Config()
- label_weights: Optional[dict[str, float]] =
None
Default JSON
{
"load_path": null,
"save_path": null,
"freeze": false,
"shared_module_key": null,
"loss": {
"CrossEntropyLoss": {}
},
"label_weights": null
}