Commit eeb43651 authored by Bharath Ramsundar's avatar Bharath Ramsundar
Browse files

Changes

parent 067cc2d8
Loading
Loading
Loading
Loading
+1 −2
Original line number Diff line number Diff line
@@ -35,8 +35,7 @@ class Featurizer(object):

    Returns
    -------
    A numpy array containing a featurized representation of
    `datapoints`.
    A numpy array containing a featurized representation of `datapoints`.
    """
    datapoints = list(datapoints)
    features = []
+3 −2
Original line number Diff line number Diff line
@@ -352,9 +352,10 @@ class ConvMolFeaturizer(MolecularFeaturizer):

  References
  ---------

  .. [1] Duvenaud, David K., et al. "Convolutional networks on graphs for
         learning molecular fingerprints." Advances in neural information processing
         systems. 2015.
         learning molecular fingerprints." Advances in neural information
         processing systems. 2015.

  Note
  ----
+12 −11
Original line number Diff line number Diff line
@@ -76,6 +76,7 @@ class SmilesToSeq(MolecularFeaturizer):

  def __init__(self, char_to_idx, max_len=250, pad_len=10, **kwargs):
    """Initialize this class. 

    Parameters
    ----------
    char_to_idx: dict