python - 从训练数据中排除某些标签但保留在测试数据中，sklearn cross_val_score

Question

我正在使用 SKLearn 训练一些模型并使用 cross_val_score() 对其进行评估。我预测一个介于 1 和 7 之间的整数值；但是，在训练期间，我只想在标签为 1、2、5、6 或 7 的样本上训练模型，但我想在整个 1-7 范围内的样本上进行测试。使用 train_test_split 很容易做到这一点；但是有谁知道我如何在 cross_val_score() 中做到这一点（即在每次迭代时从训练折叠中删除样本？）。我的问题类似于这个问题；但是，我仍然不太确定该怎么做。

标签： pythonmachine-learningscikit-learncross-validation