graph_ml.py
from torch_geometric.datasets import KarateClub