Generating regression data, fitting it, training, and testing