Regression in Python using Sklearn, XGBoost and PySpark