A simple python implementation of a restricted Boltzmann machine