SoftLabelTensorizer.Config¶

Component: SoftLabelTensorizer

class SoftLabelTensorizer.Config[source]: Bases: LabelTensorizer.Config

All Attributes (including base classes)

is_input: bool = False

column: str = 'label'

allow_unknown: bool = False

pad_in_vocab: bool = False

label_vocab: Optional[list[str]] = None

label_vocab_file: Optional[str] = None

add_labels: Optional[list[str]] = None

probs_column: str = 'target_probs'

logits_column: str = 'target_logits'

labels_column: str = 'target_labels'

Default JSON

{
    "is_input": false,
    "column": "label",
    "allow_unknown": false,
    "pad_in_vocab": false,
    "label_vocab": null,
    "label_vocab_file": null,
    "add_labels": null,
    "probs_column": "target_probs",
    "logits_column": "target_logits",
    "labels_column": "target_labels"
}