示例#1
0
def shuffle_and_split(data: RDD, fold_n: int, seed: int = 0) -> list:
    fold_weights = [1 / fold_n] * fold_n
    return data.randomSplit(fold_weights, seed)
示例#2
0
def shuffle_and_split(data: RDD, fold_n: int, seed = 0):
    fold_weights = [1 / fold_n] * fold_n
    return data.randomSplit(fold_weights)