Я использовал классификатор Случайный лес через sklearn в моем тестовом наборе данных. Когда я использую прогнозирование, случайный лес предсказывает метки классов набора тестовых данных. Я могу загрузить прогнозируемые тестовые метки в качестве выходного файла. Можно ли позволить алгоритму разделить тестовые данные после классификации, чтобы я мог загружать 2 набора данных (например, наборы данных класса A и наборы данных класса B) отдельно, а не просто загружать предсказанные метки классов?