Innocdudb的博客

原创 DataFrame转成DataLoader(数模总结)

在上周数模比赛中有遇到DataFrame类型数据，同时需要利用pytorch建立bp神经网络，而在pytroch输入的数据集需是DataLoader，在网络上查询许久找到使用方法，因此记录如下：首先是pytorch建立神经网络结构（回归模型）import torchfrom torch import nnnet=nn.Sequential(nn.Sequential(nn.Linear(20, 20),nn.Dropout(0.3),nn.Sigmoid()),nn.Sequential(nn.