Returns default metric specs for binary classification problems.
tfma.metrics.default_binary_classification_specs(
model_names: Optional[List[Text]] = None,
output_names: Optional[List[Text]] = None,
binarize: Optional[tfma.BinarizationOptions
] = None,
aggregate: Optional[tfma.AggregationOptions
] = None,
include_loss: bool = True
) -> List[tfma.MetricsSpec
]
Args |
model_names
|
Optional model names (if multi-model evaluation).
|
output_names
|
Optional list of output names (if multi-output model).
|
binarize
|
Optional settings for binarizing multi-class/multi-label metrics.
|
aggregate
|
Optional settings for aggregating multi-class/multi-label
metrics.
|
include_loss
|
True to include loss.
|