ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

Module: tfma.types

Types.

Classes

class EvalSharedModel: Shared model used during extraction and evaluation.

class FeaturesPredictionsLabels: FeaturesPredictionsLabels(input_ref, features, predictions, labels)

class MaterializedColumn: MaterializedColumn(name, value)

class ModelLoader: Model loader is responsible for loading shared model types.

class RaggedTensorValue: RaggedTensorValue encapsulates a batch of ragged tensor values.

class SparseTensorValue: SparseTensorValue encapsulates a batch of sparse tensor values.

class StructuredMetricValue: The base class for all structured metrics used within TFMA.

class ValueWithTDistribution: Represents the t-distribution value.

Type Aliases

DictOfFetchedTensorValues: The central part of internal API.

DictOfTensorType: The central part of internal API.

DictOfTensorTypeMaybeDict: The central part of internal API.

DictOfTensorValue: The central part of internal API.

DictOfTensorValueMaybeDict: The central part of internal API.

DictOfTypeSpec: The central part of internal API.

DictOfTypeSpecMaybeDict: The central part of internal API.

Extracts: The central part of internal API.

FPLKeyType: The central part of internal API.

MaybeMultipleEvalSharedModels: The central part of internal API.

MetricValueType: The central part of internal API.

MetricVariablesType: The central part of internal API.

PrimitiveMetricValueType: The central part of internal API.

TensorOrOperationType: The central part of internal API.

TensorType: The central part of internal API.

TensorTypeMaybeDict: The central part of internal API.

TensorTypeMaybeMultiLevelDict: The central part of internal API.

TensorValue: The central part of internal API.

TensorValueMaybeDict: The central part of internal API.

TensorValueMaybeMultiLevelDict: The central part of internal API.

TypeSpecMaybeDict: The central part of internal API.

TypeSpecMaybeMultiLevelDict: The central part of internal API.

ConcreteStructuredMetricValue Instance of typing.TypeVar
NumericMetricValueTypes (<class 'numbers.Number'>, <class 'numpy.ndarray'>, <class 'tensorflow_model_analysis.types.StructuredMetricValue'>)