MultiLabelOutputLayer.ConfigΒΆ
Component: MultiLabelOutputLayer
-
class
MultiLabelOutputLayer.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
}