use column name instead of class name for FID metric identification

microsoft · Jan 6, 2025 · e5b01fb · e5b01fb
1 parent db4a00d
commit e5b01fb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pe/callback/common/compute_fid.py b/pe/callback/common/compute_fid.py
@@ -47,6 +47,6 @@ def __call__(self, syn_data):
             mu2=syn_mu,
             sigma2=syn_sigma,
         )
-        metric_item = FloatMetricItem(name=f"fid_{type(self._embedding).__name__}", value=fid)
+        metric_item = FloatMetricItem(name=f"fid_{self._embedding.column_name}", value=fid)
         execution_logger.info(f"Finished computing FID ({type(self._embedding).__name__})")
         return [metric_item]