dgl.data.chem.CanonicalAtomFeaturizer

class dgl.data.chem.CanonicalAtomFeaturizer(atom_data_field='h')[source]

A default featurizer for atoms.

The atom features include:

  • One hot encoding of the atom type. The supported atom types include C, N, O, S, F, Si, P, Cl, Br, Mg, Na, Ca, Fe, As, Al, I, B, V, K, Tl, Yb, Sb, Sn, Ag, Pd, Co, Se, Ti, Zn, H, Li, Ge, Cu, Au, Ni, Cd, In, Mn, Zr, Cr, Pt, Hg, Pb.
  • One hot encoding of the atom degree. The supported possibilities include 0 - 10.
  • One hot encoding of the number of implicit Hs on the atom. The supported possibilities include 0 - 6.
  • Formal charge of the atom.
  • Number of radical electrons of the atom.
  • One hot encoding of the atom hybridization. The supported possibilities include SP, SP2, SP3, SP3D, SP3D2.
  • Whether the atom is aromatic.
  • One hot encoding of the number of total Hs on the atom. The supported possibilities include 0 - 4.
Parameters:atom_data_field (str) – Name for storing atom features in DGLGraphs, default to be ‘h’.
__init__(atom_data_field='h')[source]

Initialize self. See help(type(self)) for accurate signature.

Methods

__init__([atom_data_field]) Initialize self.

Attributes

feat_size Returns feature size