MultiLabelClassificationMetricReporter.Config

Component: MultiLabelClassificationMetricReporter

class MultiLabelClassificationMetricReporter.Config

Bases: ClassificationMetricReporter.Config

All Attributes (including base classes)

output_path: str = '/tmp/test_out.txt'

pep_format: bool = False

student_column_names: list[str] = []

log_gradient: bool = False

model_select_metric: ComparableClassificationMetric = <ComparableClassificationMetric.ACCURACY: 'accuracy'>

target_label: Optional[str] = None

text_column_names: list[str] = ['text']

additional_column_names: list[str] = []

recall_at_precision_thresholds: list[float] = [0.2, 0.4, 0.6, 0.8, 0.9]

is_memory_efficient: bool = False

Default JSON

{
    "output_path": "/tmp/test_out.txt",
    "pep_format": false,
    "student_column_names": [],
    "log_gradient": false,
    "model_select_metric": "accuracy",
    "target_label": null,
    "text_column_names": [
        "text"
    ],
    "additional_column_names": [],
    "recall_at_precision_thresholds": [
        0.2,
        0.4,
        0.6,
        0.8,
        0.9
    ],
    "is_memory_efficient": false
}