covariate-shift-adaption An example of using a discriminator to correct for a difference in the distributions between the training and test data. Please see the Jupyter notebook here.