Module: tfma

Defined in __init__.py.

Init module for TensorFlow Model Analysis.

Modules

constants module: Constants used in TensorFlow Model Analysis.

evaluators module: Init module for TensorFlow Model Analysis evaluators.

export module: Library for exporting the EvalSavedModel.

exporter module: Exporter class represents different flavors of model export.

extractors module: Init module for TensorFlow Model Analysis extractors.

post_export_metrics module: Library containing helpers for adding post export metrics for evaluation.

types module: Types.

validators module: Init module for TensorFlow Model Analysis validators.

version module: Contains the version string for this release of TFMA.

view module: Initializes TFMA's view rendering api.

writers module: Init module for TensorFlow Model Analysis writers.

Classes

class EvalConfig: Config used for extraction and evaluation.

class EvalResult: EvalResult(slicing_metrics, plots, config)

class EvalSharedModel: Shared model used during extraction and evaluation.

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

class MaterializedColumn: MaterializedColumn(name, value)

Functions

ExtractAndEvaluate(...): Performs Extractions and Evaluations in provided order.

ExtractEvaluateAndWriteResults(...): PTransform for performing extraction, evaluation, and writing results.

InputsToExtracts(...): Converts serialized inputs (e.g. examples) to Extracts.

Validate(...): Performs validation of alternative evaluations.

WriteResults(...): Writes Evaluation or Validation results using given writers.

compound_key(...): Returns a compound key based on a list of keys.

create_keys_key(...): Creates secondary key representing the sparse keys associated with key.

create_values_key(...): Creates secondary key representing sparse values associated with key.

default_eval_shared_model(...): Returns default EvalSharedModel.

default_evaluators(...): Returns the default evaluators for use in ExtractAndEvaluate.

default_extractors(...): Returns the default extractors for use in ExtractAndEvaluate.

default_writers(...): Returns the default writers for use in WriteResults.

load_eval_result(...): Creates an EvalResult object for use with the visualization functions.

load_eval_results(...): Run model analysis for a single model on multiple data sets.

make_eval_results(...): Run model analysis for a single model on multiple data sets.

multiple_data_analysis(...): Run model analysis for a single model on multiple data sets.

multiple_model_analysis(...): Run model analysis for multiple models on the same data set.

run_model_analysis(...): Runs TensorFlow model analysis.

unique_key(...): Returns a unique key given a list of current keys.

Other Members

ANALYSIS_KEY

ATTRIBUTIONS_KEY

BASELINE_KEY

BASELINE_SCORE_KEY

CANDIDATE_KEY

DATA_CENTRIC_MODE

EXAMPLE_SCORE_KEY

FEATURES_KEY

FEATURES_PREDICTIONS_LABELS_KEY

INPUT_KEY

LABELS_KEY

METRICS_KEY

MODEL_CENTRIC_MODE

PLOTS_KEY

PREDICTIONS_KEY

VERSION_STRING