Загрузка набора данных SNAP в объект torch_geometric.data.Data.

Я пытаюсь загрузить набор данных SNAP (на данный момент ego-facebook) в объект. У меня проблемы.

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

0 ответов

Другие вопросы по тегам