python linear regression correlation coefficient