tfma.default_writers

View source on GitHub

Returns the default writers for use in WriteResults.

Note, sharding will be enabled by default if an output_file_format is provided. Filenames will be -SSSSS-of-NNNNN. where SSSSS is the shard number and NNNNN is the number of shards.

output_path Output path.
eval_shared_model Optional shared model (single-model evaluation) or list of shared models (multi-model evaluation). Only required if legacy add_metrics_callbacks are used.
eval_config Optional eval config for writing out config along with results.
display_only_data_location Optional path indicating where the examples were read from. This is used only for display purposes - data will not actually be read from this path.
display_only_data_file_format Optional format of the input examples. This is used only for display purposes.
output_file_format File format to use when saving files. Currently only 'tfrecord' is supported.
add_metric_callbacks Optional list of metric callbacks (if used).