From e5b01fbb922df6699942865436200b48e55a4b2e Mon Sep 17 00:00:00 2001 From: Zinan Lin Date: Mon, 6 Jan 2025 14:45:14 -0800 Subject: [PATCH] use column name instead of class name for FID metric identification --- pe/callback/common/compute_fid.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pe/callback/common/compute_fid.py b/pe/callback/common/compute_fid.py index 8d3dd6d..a584c9a 100644 --- 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]