modify trainNN to accept initialTheta, and number of iterations.
type signatures and comments
predictLikelihood function from theta and x