neural network regression python github