polynomial regression machine learning code