Pertanyaan yang diberi tag scikit-learn

95
Latihan Stratifikasi / Uji-split dalam scikit-learn

Saya perlu membagi data saya menjadi satu set pelatihan (75%) dan set pengujian (25%). Saat ini saya melakukannya dengan kode di bawah ini: X, Xt, userInfo, userInfo_train = sklearn.cross_validation.train_test_split(X, userInfo) Namun, saya ingin membuat stratifikasi set data pelatihan saya....