Dependencies: numpy, matplotlib, torchvision, sklearn

Figure 1 in the paper is generated by running the provided code with 
K = 10, 25, 50, 100 (set on line 174). The resulting charts were 
lightly edited (by hand) for presentation in the paper. 