![]() |
Confusion matrix plot.
Inherits From: Metric
tfma.metrics.ConfusionMatrixPlot(
num_thresholds: int = DEFAULT_NUM_THRESHOLDS,
name: str = CONFUSION_MATRIX_PLOT_NAME
)
Args | |
---|---|
num_thresholds
|
Number of thresholds to use when discretizing the curve. Values must be > 1. Defaults to 1000. |
name
|
Metric name. |
Methods
computations
computations(
eval_config: Optional[tfma.EvalConfig
] = None,
schema: Optional[schema_pb2.Schema] = None,
model_names: Optional[List[str]] = None,
output_names: Optional[List[str]] = None,
sub_keys: Optional[List[Optional[SubKey]]] = None,
aggregation_type: Optional[AggregationType] = None,
class_weights: Optional[Dict[int, float]] = None,
example_weighted: bool = False,
query_key: Optional[str] = None
) -> tfma.metrics.MetricComputations
Creates computations associated with metric.
get_config
get_config() -> Dict[str, Any]
Returns serializable config.