scikit-learn/scikit-learn — pull request #1375 by pzc (2012-11-15 17:26:55+00:00)
Title: imbalanced data set / stratify train_set, undersample one class

Dear SciKitters,

given a data set with with 2 y_labels (80 % "0" and 20 % "1") I would like to create train set which shows an equal weighting.
Currently, train_test_split only allows for a test_training split which gives the same weighting.

From the mailing list, I got the recommendation to submit a PR.

Cheers,
Paul