[SPARK-2612] [mllib] Fix data skew in ALS