Restricted Boltzmann Machine Python