题解|交叉验证数据拆分

交叉验证数据拆分

https://www.nowcoder.com/practice/90bc72a0f81e41d39cf4d0488bbfc8a0?tpId=377&tags=&title=&difficulty=0&judgeStatus=0&rp=0&sourceUrl=%2Fexam%2Foj

交叉验证是一种评估机器学习模型性能的方法,它通过将数据集分为训练集和测试集,来评估模型的性能。 其目的在于评估模型在不同数据集上的表现,以确保模型的泛化能力;并且在数据量较少时,可以有效地评估模型的性能。

本题中要求了将数据集分为k折,并且要求了随机种子,但是只需要在采样之前进行一次随机打乱即可。

标准代码如下

def cross_validation_split(data, k, seed=42):
    np.random.seed(seed)
    np.random.shuffle(data)

    n, m = data.shape
    sub_size = int(np.ceil(n / k))
    id_s = np.arange(0, n, sub_size)
    id_e = id_s + sub_size
    if id_e[-1] > n: id_e[-1] = n

    return [[np.concatenate([data[: id_s[i]], data[id_e[i]:]], axis=0).tolist(), data[id_s[i]: id_e[i]].tolist()] for i in range(k)]
全部评论

相关推荐

野猪不是猪🐗:这种直接口头上答应,骗面试,面完了直接拉黑,相当于给自己攒面经了(
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客企业服务