7
Стратифікований поїзд / тест-спліт у науковому дослідженні
Мені потрібно розділити свої дані на навчальний набір (75%) та набір тестів (25%). На даний момент я роблю це за допомогою коду нижче: X, Xt, userInfo, userInfo_train = sklearn.cross_validation.train_test_split(X, userInfo) Однак я хотів би розшарувати свій навчальний набір даних. Як це зробити? Я досліджував StratifiedKFoldметод, але не дозволяв мені конкретизувати …