Fix code format.
Increase parallel by grouping weightedDataSet into RDD of multiple arrays.