Spark train test split
<p>I am curious if there is something similar to sklearn's  <a href="http://scikit-learn.org/stable/modules/generated/sklearn.model_selection.StratifiedShuffleSplit.html" rel="noreferrer">http://scikit-learn.org/stable/modules/generated/sklearn.model_selection.StratifiedShuffleSplit.html</a> for apache-spark in the latest 2.0.1 release.</p>

<p>So far I could only find <a href="https://spark.apache.org/docs/latest/mllib-statistics.html#stratified-sampling" rel="noreferrer">https://spark.apache.org/docs/latest/mllib-statistics.html#stratified-sampling</a> which does not seem to be a great fit for splitting heavily imbalanced dataset into train /test samples.</p>