The code has been tested running under Python 3.7.4, with the following packages and their dependencies installed: numpy==1.16.5 matplotlib==3.1.0 sklearn==0.21.3 seaborn==0.11.2 pytorch==1.7.1 ...