agmm.MMDGMM

class agmm.MMDGMM(learner, adversary_g, n_samples, kernel, sigma)[source]

AGMM with Maximum Mean Discrepancy.

Parameters

  • learner – a pytorch neural net module for the learner.

  • adversary_g – a pytorch neural net module for the g function of the adversary.

  • n_samples – number of samples.

  • kernel – the kernel function.

  • sigma – float corresponding to the precision of the kernel.