首页 > 解决方案 > Pytorch Geometric:如何从 pytorch 几何数据对象列表在 colab 笔记本中创建一个小型临时数据集

问题描述

我正在使用 Colab 工作表并生成了 Pytorch 几何数据对象的 python 列表。我现在想把它们变成一个数据集,只在这个笔记本中使用。我怎样才能做到这一点?现有文档似乎面向长期数据集。

当我使用标准 Pytorch 时,我使用了 torch.FloatTensor() 和 TensorDataset() 的组合来创建我自己的数据集以用于随机拆分。

标签: pytorchdatasetpytorch-geometric

解决方案


推荐阅读