小编M. *_*bio的帖子

将Pandas数据帧转换为PyTorch张量?

我想使用个人数据库在PyTorch上训练一个简单的神经网络.此数据库从Excel文件导入并存储在df.

其中一列被命名"Target",它是网络的目标变量.如何使用此数据框作为PyTorch神经网络的输入?

我试过这个,但它不起作用:

target = pd.DataFrame(data = df['Target'])
train = data_utils.TensorDataset(df, target)
train_loader = data_utils.DataLoader(train, batch_size = 10, shuffle = True)
Run Code Online (Sandbox Code Playgroud)

python dataframe pandas pytorch

19
推荐指数
6
解决办法
2万
查看次数

标签 统计

dataframe ×1

pandas ×1

python ×1

pytorch ×1