agmm.KernelLossAGMM

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

AGMM with kernel loss.

Parameters

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

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

  • kernel – the kernel function.

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