自定义折叠用于scikit-learn中的交叉验证

Question

我想将GridSearchCV（带有n_jobs＆gt; 1）用于特定的分类器，但我有关于来自其他来源的10倍交叉验证的折叠的信息。有没有办法输入已分成折叠的数据，而不是使用GridSearchCV创建的折叠。

谢谢！

Answer 1

您可以创建一个自定义CV迭代器，例如LeaveOneLabelOut或LeavePLabelOut，并将您自己的预先计算的折叠编码为整数数组，然后将该CV迭代器作为cv参数传递。 cross_val_score和GridSearchCV实用程序。