Two Sigma interview question

linear regression coding test