Questions marquées «scikit-learn»

88
Train stratifié / Test-split dans scikit-learn

Je dois diviser mes données en un ensemble d'entraînement (75%) et un ensemble de test (25%). Je le fais actuellement avec le code ci-dessous: X, Xt, userInfo, userInfo_train = sklearn.cross_validation.train_test_split(X, userInfo) Cependant, j'aimerais stratifier mon ensemble de données...