Large prediction results unless using repartition(1) in databricks with lgbm model in mmlspark

Question

I\'m using mmlspark lgbm model for regression problem and faced strange thing. If using all normal code as in the example, results will be terrible, becouse predictions are huge