Загрузка набора данных SNAP в объект torch_geometric.data.Data.
Я пытаюсь загрузить набор данных SNAP (на данный момент ego-facebook) в объект. У меня проблемы.
- Существует ли какая-либо стандартная библиотека/метод для загрузки набора данных SNAP в
torch_geometric.data.Data
объект? - Я попробовал использовать
torch_geometric.datasets.SNAPDataset
. Это дает мне хранилище 10 данных. Как я могу объединить в один объект, чтобы я мог создать одинNetworkx
объект? - Я попытался создать данные, добавив каждое ребро. Позже я хотел использовать
torch_geometric.transforms.RandomNodeSplit
для создания масок. Но это не добавило никакой маски. Я полагалRandomNodeSplit
проходитdata.node_stores
. В данном случае у меня не былоnode_stores
в моемData
объект. Как правильно создать объект данных и добавить к нему узлы/ребра, чтобы он соответствовалtorch_geometric
стиль?