Preguntas etiquetadas con scikit-learn

88
Tren estratificado / Test-split en scikit-learn

Necesito dividir mis datos en un conjunto de entrenamiento (75%) y un conjunto de prueba (25%). Actualmente hago eso con el siguiente código: X, Xt, userInfo, userInfo_train = sklearn.cross_validation.train_test_split(X, userInfo) Sin embargo, me gustaría estratificar mi conjunto de datos de...