Ввод графика (networkx) для Dataloader в Pytorch - PullRequest
0 голосов
/ 03 марта 2020

Я новичок ie в Графике, используя глубокое обучение. Я пытаюсь создать DataLoader для Pytorch для обучения.

Я использую networkx для сохранения своего графика, затем получаю dataset из утилиты torch_geometri c, чтобы иметь набор данных для DataLoader. Но я не могу тренировать свою модель. Мой код:

#import...
G = nx.karate_club_graph()
dataset = torch_geometric.utils.from_networkx(G)
dataloader = DataLoader(dataset=dataset, batch_size=32, shuffle=True)

for data in dataloader:
   print(data) # Error here: TypeError: getattr(): attribute name must be string

Итак, как мне поставить график ввода для Pytorch для обучения?

...