Тренировка мл моделей на свечах на перегородках. Такой, что будет обученная модель для каждого раздела dataframe

Question

Как провести параллельное обучение модели на раздел в spark, используя scala? Решение, данное здесь, находится в Pyspark. Я ищу решение в scala. Как эффективно построить одну модель ML на раздел в Spark с помощью foreachPartition?