dgllife.utils.atom_total_num_H_one_hot

dgllife.utils.atom_total_num_H_one_hot(atom, allowable_set=None, encode_unknown=False)[source]

One hot encoding for the total number of Hs of an atom.

Parameters
  • atom (rdkit.Chem.rdchem.Atom) – RDKit atom instance.

  • allowable_set (list of int) – Total number of Hs to consider. Default: 0 - 4.

  • encode_unknown (bool) – If True, map inputs not in the allowable set to the additional last element. (Default: False)

Returns

List of boolean values where at most one value is True.

Return type

list