你如何在 ML 中實現 make_folds function

Question

如何在 ML 中實現 make_folds function

def make_folds(x, num_folds):
    """ Divides the array `x` along axis-0 into a list of equal-sized 
    sub-arrays.

x 是 numpy 數組

Answer 1

你可以嘗試這樣的事情：

def make_folds(x, num_folds):
    """ Divides the array `x` along axis-0 into a list of equal-sized 
    sub-arrays."""
    assert len(x)%num_folds == 0, "Given array can not be split into {} equal-size subarrays".format(num_folds)
    return np.split(x, num_folds)

或直接使用https://scikit-learn.org/stable/modules/generated/sklearn.model_selection.KFold.html如果您想進行 Kfold 驗證