tfma.metrics.SubKey

View source on GitHub

A SubKey identifies a sub-types of metrics and plots.

Only one of class_id, k, or top_k can be set at a time.

class_id Used with multi-class metrics to identify a specific class ID.
k Used with multi-class metrics to identify the kth predicted value.
top_k Used with multi-class and ranking metrics to identify top-k predicted values.

Methods

from_proto

View source

Creates class from proto.

to_proto

View source

Converts key to proto.