pe.callback.common.compute_fid module

class pe.callback.common.compute_fid.ComputeFID(priv_data, embedding, filter_criterion=None)[source]

Bases: Callback

The callback that computes the Frechet Inception Distance (FID) between the private and synthetic data.

__call__(syn_data)[source]

This function is called after each PE iteration that computes the FID between the private and synthetic data.

Parameters:

syn_data (pe.data.Data) – The synthetic data

Returns:

The FID between the private and synthetic data

Return type:

list[pe.metric_item.FloatMetricItem]

__init__(priv_data, embedding, filter_criterion=None)[source]

Constructor.

Parameters:

  • priv_data (pe.data.Data) – The private data

  • embedding (pe.embedding.Embedding) – The embedding to compute the FID

  • filter_criterion (dict, optional) – Only computes the metric based on samples satisfying the criterion. None means no filtering. Defaults to None