В зависимости от вашей цели вы можете использовать метод cross_validation, и он автоматически выполнит разделение за вас. Пример: cross_validate(algorithm_to_use, data, measures=['RMSE'], cv=5)... cv=5 сообщает итератору перекрестной проверки, как разделить данные.
См. Документацию по Перекрестная проверка .
Вы также можете изучить пакет выбора модели для того, что вам нужно .
Контекст важно, когда вы пытаетесь помочь вам найти решение.