Perguntas com a marcação «scikit-learn»

88
Treino estratificado / divisão de teste no scikit-learn

Preciso dividir meus dados em um conjunto de treinamento (75%) e um conjunto de teste (25%). Atualmente faço isso com o código abaixo: X, Xt, userInfo, userInfo_train = sklearn.cross_validation.train_test_split(X, userInfo) No entanto, gostaria de estratificar meu conjunto de dados de...