Commit 22933d68 authored by alat-rights's avatar alat-rights
Browse files

Documentation change

parent 77695d5f
Loading
Loading
Loading
Loading
+3 −3
Original line number Diff line number Diff line
@@ -19,13 +19,13 @@ ZINC_CHARSET = [


class OneHotFeaturizer(MolecularFeaturizer):
  """Encodes SMILES as a one-hot array.
  """Encodes SMILES or any arbitrary string as a one-hot array.

  This featurizer encodes SMILES string as a one-hot array.
  This featurizer encodes a string or any arbitrary string as a one-hot array.

  Note
  ----
  This class requires RDKit to be installed.
  This class requires RDKit to be installed to work with RDKit molecules.
  """

  def __init__(self, charset: List[str] = ZINC_CHARSET, max_length: int = 100):