Commit d7f98f96 authored by yurievnamaria's avatar yurievnamaria
Browse files

added to description of convmolfeaturizer

parent a1ec4f3f
Loading
Loading
Loading
Loading
+0 −17
Original line number Diff line number Diff line
import os
import tempfile
import deepchem as dc


def test_singleton_csv_fragment_load_with_per_atom_fragmentation():
  """Test a case where special form of  dataaset is created from csv:
   dataset of fragments of molecules  for subsequent model interpretation """
  with tempfile.NamedTemporaryFile(mode='w', delete=False) as fin:
    fin.write("smiles,endpoint\nc1ccccc1,1")
  featurizer = dc.feat.ConvMolFeaturizer(per_atom_fragmentation=True)
  tasks = ["endpoint"]
  loader = dc.data.CSVFragmentLoader(
      tasks=tasks, feature_field="smiles", featurizer=featurizer)
  X = loader.create_dataset(fin.name)
  assert len(X) == 6
  os.remove(fin.name)
+7 −1
Original line number Diff line number Diff line
@@ -632,13 +632,19 @@ class ConvMolFeaturizer(MolecularFeaturizer):

  Examples
  ---------
  Using ConvMolFeaturizer to create featurized fragments derived from molecules of interest
  Using ConvMolFeaturizer to create featurized fragments derived from molecules of interest.
  This is used only in the context of performing interpretation of models using atomic
  contributions (atom-based model interpretation)
  >>> import deepchem as dc
  >>> smiles = ["C", "CCC"]
  >>> featurizer=dc.feat.ConvMolFeaturizer(per_atom_fragmentation=True)
  >>> f = featurizer.featurize(smiles)
  >>> len(f) # contains 2 lists with  featurized fragments from 2 mols

  See Also
  --------
  Detailed examples of `GraphConvModel` interpretation are provided in Tutorial #28

  References
  ---------

+0 −1
Original line number Diff line number Diff line
import tempfile
import os
import numpy as np
import deepchem as dc