Simple example of how to train a neural network to learn to weight data to follow a given probability distribution. This example is written on a Jupyter notebook and uses PyTorch and NumPy.
Follow the link to find the repository and instructions: https://github.com/jbossios/simple-ml-reweighting-example