蓝桥云课xBilibili编程竞赛6月专场【4.分而治之】题解

最新推荐文章于 2024-07-10 19:27:42 发布

qq_43335313

最新推荐文章于 2024-07-10 19:27:42 发布

阅读量65

点赞数

分类专栏：机器学习入门文章标签： python 机器学习人工智能算法

本文链接：https://blog.csdn.net/qq_43335313/article/details/125597537

版权

机器学习入门专栏收录该内容

5 篇文章 0 订阅

订阅专栏

在这里插入图片描述

题解

from typing import List


def data_pipeline(data: List[List[int]], batch_size: int) -> List[List[int]]:
    # 在此函数中完成 mini-batch
    
    for i in range(0,len(data),batch_size):
        batch_data = []
        print(i)
        for k in range(batch_size):
            if i+k >= len(data):
                break
            else:
                batch_data.append(data[i+k])
        yield batch_data
data = [[1, 2],
        [1, 3],
        [3, 5],
        [2, 1],
        [3, 3]]
batch_size = 2
for i in data_pipeline(data, batch_size):
    print(i)