input for regression and test is now just the matrix (and theta for test)