tfma.metrics.ConfusionMatrixPlot

Stay organized with collections Save and categorize content based on your preferences.

Confusion matrix plot.

Inherits From: Metric

tfma.metrics.ConfusionMatrixPlot(
    num_thresholds: int = DEFAULT_NUM_THRESHOLDS,
    name: str = CONFUSION_MATRIX_PLOT_NAME,
    **kwargs
)

num_thresholds Number of thresholds to use when discretizing the curve. Values must be > 1. Defaults to 1000.
name Metric name.
**kwargs (Optional) Additional args to pass along to init (and eventually on to _confusion_matrix_plot). These kwargs are useful for subclasses to pass information from their init to the create_computation_fn.

compute_confidence_interval Whether to compute confidence intervals for this metric.

Note that this may not completely remove the computational overhead involved in computing a given metric. This is only respected by the jackknife confidence interval method.

Methods

computations

View source

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

View source

get_config() -> Dict[str, Any]

Returns serializable config.