https://medium.com/@ebojacky/machine-learning-using-spark-ml-do-regression-classification-and-clustering-with-spark-ml-b6b32fa54d87