コード例 #1
0
ファイル: hold_out.py プロジェクト: AlexFridman/code_temp 
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)