标签： scikit-learn classification
我在测试数据集上通过sklearn使用了随机森林分类器。当我使用预测时，随机森林会预测测试数据集的类标签。我可以将预测的测试标签下载为输出文件。有什么方法可以让算法对分类后的测试数据进行划分，以便我可以分别下载2个数据集（例如A类数据集和B类数据集），而不仅仅是下载预测的类标签？