I\'m trying to create the equivalent of the following stratified sampling in scala.
Python code which is working as expected
stratifiedDF = labeledAllD
