QM9Dataset

class dgl.data.QM9Dataset(label_keys, cutoff=5.0, raw_dir=None, force_reload=False, verbose=False, transform=None)

Bases: DGLDataset

QM9 dataset for graph property prediction (regression)

This dataset consists of 130,831 molecules with 12 regression targets. Nodes correspond to atoms and edges correspond to close atom pairs.

This dataset differs from QM9EdgeDataset in the following aspects:

1. Edges in this dataset are purely distance-based.

2. It only provides atoms’ coordinates and atomic numbers as node features

3. It only provides 12 regression targets.

Reference:

• “Quantum-Machine.org”,

• “Directional Message Passing for Molecular Graphs”

Statistics:

• Number of graphs: 130,831

• Number of regression targets: 12

Keys	Property	Description	Unit
mu	$\mu$	Dipole moment	$\text{D}$
alpha	$\alpha$	Isotropic polarizability	${a_0}^3$
homo	${ϵ}_{\text{HOMO}}$	Highest occupied molecular orbital energy	$\text{eV}$
lumo	${ϵ}_{\text{LUMO}}$	Lowest unoccupied molecular orbital energy	$\text{eV}$
gap	$\mathrm{\Delta }ϵ$	Gap between ${ϵ}_{\text{HOMO}}$ and ${ϵ}_{\text{LUMO}}$	$\text{eV}$
r2	$⟨R^2⟩$	Electronic spatial extent	${a_0}^2$
zpve	$\text{ZPVE}$	Zero point vibrational energy	$\text{eV}$
U0	$U_0$	Internal energy at 0K	$\text{eV}$
U	$U$	Internal energy at 298.15K	$\text{eV}$
H	$H$	Enthalpy at 298.15K	$\text{eV}$
G	$G$	Free energy at 298.15K	$\text{eV}$
Cv	$c_{\text{v}}$	Heat capavity at 298.15K	$\frac{\text{cal}}{\text{mol K}}$

Parameters:

• label_keys (list) – Names of the regression property, which should be a subset of the keys in the table above.

• cutoff (float) – Cutoff distance for interatomic interactions, i.e. two atoms are connected in the corresponding graph if the distance between them is no larger than this. Default: 5.0 Angstrom

• raw_dir (str) – Raw file directory to download/contains the input data directory. Default: ~/.dgl/

• force_reload (bool) – Whether to reload the dataset. Default: False

• verbose (bool) – Whether to print out progress information. Default: True.

• transform (callable, optional) – A transform that takes in a DGLGraph object and returns a transformed version. The DGLGraph object will be transformed before every access.

num_tasks

Number of prediction tasks

Type:

int

num_labels

(DEPRECATED, use num_tasks instead) Number of prediction tasks

Type:

int

Raises:

UserWarning – If the raw data is changed in the remote server by the author.

Examples

>>> data = QM9Dataset(label_keys=['mu', 'gap'], cutoff=5.0)
>>> data.num_tasks
2
>>>
>>> # iterate over the dataset
>>> for g, label in data:
...     R = g.ndata['R'] # get coordinates of each atom
...     Z = g.ndata['Z'] # get atomic numbers of each atom
...     # your code here...
>>>

__getitem__(idx)

Get graph and label by index

Parameters:

idx (int) – Item index

Returns:

• dgl.DGLGraph – The graph contains:

  • ndata['R']: the coordinates of each atom

  • ndata['Z']: the atomic number

• Tensor – Property values of molecular graphs

__len__()

Number of graphs in the dataset.

Return type:

int