7
Stratified Train / Test-split ใน scikit-learn
ฉันต้องการแบ่งข้อมูลของฉันออกเป็นชุดฝึก (75%) และชุดทดสอบ (25%) ขณะนี้ฉันใช้รหัสด้านล่าง: X, Xt, userInfo, userInfo_train = sklearn.cross_validation.train_test_split(X, userInfo) อย่างไรก็ตามฉันต้องการแบ่งชั้นชุดข้อมูลการฝึกอบรมของฉัน ฉันจะทำอย่างไร? ฉันได้ตรวจสอบStratifiedKFoldวิธีการนี้แล้ว แต่อย่าให้ฉันระบุการแบ่ง 75% / 25% และแบ่งชั้นชุดข้อมูลการฝึกอบรมเท่านั้น