Github user mengxr commented on a diff in the pull request:

    https://github.com/apache/spark/pull/18#discussion_r11457437
  
    --- Diff: mllib/src/main/scala/org/apache/spark/mllib/util/MLUtils.scala ---
    @@ -172,6 +182,20 @@ object MLUtils {
       }
     
       /**
    +   * Return a k element list of pairs of RDDs with the first element of 
each pair
    +   * containing a unique 1/Kth of the data and the second element contain 
the compliment of that.
    +   */
    +  def kFold[T : ClassTag](rdd: RDD[T], numFolds: Int, seed: Int): 
List[Pair[RDD[T], RDD[T]]] = {
    --- End diff --
    
    About the return type, it is the same as `List[(RDD[T], RDD[T])]`. The 
latter form is more common in Spark codebase. Also, can we change the return 
type to `Array[(RDD[T], RDD[T])]`, which is Java-friendly and there is no 
overhead because `numFolds` is small.


---
If your project is set up for it, you can reply to this email and have your
reply appear on GitHub as well. If your project does not have this feature
enabled and wishes so, or if the feature is enabled but not working, please
contact infrastructure at [email protected] or file a JIRA ticket
with INFRA.
---

Reply via email to