title: Make data balanced after train test split operation (scikit)?
body: <p>I have got the problem, that after splitting my data into training and test data one class is totally missing in my test set.</p>
  
  <pre><code>Example on 60/40 split: &lt;br/&gt;
  Training: 'Orange', 0,0,0, 'Orange' &lt;br/&gt;
  Testdata: 0,0,0,0,0 
  </code></pre>
  
  <p>Obviously the word "orange" is not included in the test set. How can one ensure that the split is considering that at least some target samples are included in the test set as well as in the training set? I thought the stratify parameter would do this, but unfortunately does not it.</p>