tfma.exporter.adapt_to_remove_metrics

View source on GitHub

Modifies the given exporter to remove metrics before export.

This is useful for when you use py_func, streaming metrics, or other metrics incompatible with TFMA in your trainer. To keep these metrics in your trainer (so they still show up in Tensorboard) and still use TFMA, you can call adapt_to_remove_metrics on your TFMA exporter.

exporter Exporter to modify. Will be mutated in place.
metrics_to_remove List of names of metrics to remove.

The mutated exporter, which will be modified in place. We also return it so that this can be used in an expression.