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.

class VarLenTensorValue: VarLenTensorValue encapsulates a batch of varlen dense tensor values.

Type Aliases

AddMetricsCallbackType

DictOfFetchedTensorValues

DictOfTensorType

DictOfTensorTypeMaybeDict

DictOfTensorValue

DictOfTensorValueMaybeDict

DictOfTypeSpec

DictOfTypeSpecMaybeDict

Extracts

FPLKeyType

MaybeMultipleEvalSharedModels

MetricValueType

MetricVariablesType

PrimitiveMetricValueType

TensorOrOperationType

TensorType

TensorTypeMaybeDict

TensorTypeMaybeMultiLevelDict

TensorValue

TensorValueMaybeDict

TensorValueMaybeMultiLevelDict

TypeSpecMaybeDict

TypeSpecMaybeMultiLevelDict

ConcreteStructuredMetricValue Instance of typing.TypeVar