Pyspark CrossValidator子模型和数据集的折叠

Pyspark CrossValidator子模型和数据集的折叠,pyspark,Pyspark,在使用CollectSubModels=True运行CrossValidator之后,我想使用左侧的子模型进行预测。但是,我无法找到创建了K折叠的行的索引 有没有一种方法可以访问折叠(比如说,有一列指示行属于从1到K的哪个折叠。) 我可以手动拆分数据并进行K折叠,但我希望有一种方法不用编写自己的交叉验证程序,谢谢

在使用CollectSubModels=True运行CrossValidator之后,我想使用左侧的子模型进行预测。但是,我无法找到创建了K折叠的行的索引

有没有一种方法可以访问折叠(比如说,有一列指示行属于从1到K的哪个折叠。)

我可以手动拆分数据并进行K折叠,但我希望有一种方法不用编写自己的交叉验证程序,谢谢