Skip to content

Instantly share code, notes, and snippets.

@sidneyarcidiacono

sidneyarcidiacono/PROTEINS_embedding.py

Created May 17, 2021 20:54
Show Gist options
  • Save sidneyarcidiacono/36dd698322335d8ce6458193ca8e9852 to your computer and use it in GitHub Desktop.
Save sidneyarcidiacono/36dd698322335d8ce6458193ca8e9852 to your computer and use it in GitHub Desktop.
Download ZIP
Download dataset from pytorch-geometric
Raw
PROTEINS_embedding.py
import torch
from torch_geometric.datasets import TUDataset
# Like Spektral, pytorch geometric provides us with benchmark TUDatasets
dataset = TUDataset(root='data/TUDataset', name='PROTEINS')
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment