GennItsuka的博客

原创 简单的线性回归模型代码

get_data与get_label分别是从传入参数data以及label中取某一组的对应数据，batchsize是每次所取的数据点数，get_indices是当前批次的索引，此时传入的data和label分别是X和Y,则会每次去16个X与Y的张量进行操作get_data形状为16*4，get_label形状为16*1。yield是有存档点的特殊的return，每次退出后可保存函数内个数值的数据。2.批量生成数据的函数，每次访问该函数就会提供一些批量的数据。3.回归模型的预测值。5.随机梯度下降算法。