Skip to content

Instantly share code, notes, and snippets.

@sidneyarcidiacono

sidneyarcidiacono/PROTEINS_Embedding.py

Last active May 17, 2021 20:49
Show Gist options
  • Save sidneyarcidiacono/2bad56b694c55469954b08ef951d7c88 to your computer and use it in GitHub Desktop.
Save sidneyarcidiacono/2bad56b694c55469954b08ef951d7c88 to your computer and use it in GitHub Desktop.
Download ZIP
reading in the PROTEINS dataset
Raw
PROTEINS_Embedding.py
# Reading in the PROTEINS dataset
from spektral.datasets import TUDataset
# Spectral provides the TUDataset class, which contains benchmark datasets for graph classification
data = TUDataset('PROTEINS')
data
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment