We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 958a0e7 commit b68b4daCopy full SHA for b68b4da
tensorflow_model_analysis/metrics/tf_metric_wrapper.py
@@ -267,10 +267,12 @@ def _metric_keys_and_configs(
267
example_weighted=example_weighted))
268
if isinstance(metric, tf.keras.metrics.Metric):
269
metric_configs[updated_sub_key][output_name].append(
270
- metric_util.serialize_metric(metric))
+ metric_util.serialize_metric(metric, use_legacy_format=True)
271
+ )
272
elif isinstance(metric, tf.keras.losses.Loss):
273
loss_configs[updated_sub_key][output_name].append(
- metric_util.serialize_loss(metric))
274
+ metric_util.serialize_loss(metric, use_legacy_format=True)
275
276
return metric_keys, metric_configs, loss_configs
277
278
0 commit comments