Skip to content

Instantly share code, notes, and snippets.

@hengruizhang98
hengruizhang98 / qm9_edge.py
Last active July 29, 2021 13:49
Preprocessing QM9Edge Dataset
"""
Preprocessing QM9Edge dataset for dgl.data.QM9EdgeDatset.
When procssing this dataset, we partially refer to the implementation from https://github.com/Jack-XHP/DGL_QM9EDGE
"""
import os
import numpy as np
from dgl.data import DGLDataset
from dgl.data.utils import download, extract_archive, _get_dgl_url