Как мы делим стратифицированное разделение поезда-теста на кадрах данных Spark?

Question

Я хочу разделить фрейм данных Spark, который имеет дисбаланс классов, поэтому я хочу разделить на обучающие и тестировать фреймы данных с правильными соотношениями классов. Как это сделать на фреймах данных Spark (без использования rdd).