Spark ML TrainValidationSplit（定義済み列車と検証セットあり）

Question

Spark MLクラスのTrainValidationSplitを使用すると、setTrainRatioを設定できます。しかし、列車とテストセットを手動で作成したいと思います。 TrainValidationSplitで列車とテストセットを指定する方法はありますか？

私はScala ML APIを使用しています。

ありがとうございます！

Answer 1

申し訳ありません。少なくともスパークで

は電車と検証データの定義はrandomSplitを介して行われていること、ハードコーディングされて2.1.1：

https://github.com/apache/spark/blob/branch-2.1/mllib/src/main/scala/org/apache/spark/ml/tuning/TrainValidationSplit.scala#L100-L101

おそらくTrainValidationClassから継承し、フィット機能を拡張することができあなたが合っているように。