Commit b652d440 authored by Bharath Ramsundar's avatar Bharath Ramsundar Committed by GitHub
Browse files

Merge pull request #212 from apappu97/master

StratifiedSplitter class for Deepchem Splits
parents e578ed10 0a110a89
Loading
Loading
Loading
Loading
+17 −1
Original line number Diff line number Diff line
@@ -73,3 +73,19 @@ class TestDatasetAPI(TestAPI):
        featurizer=featurizer,
        verbosity="low")
    return loader.featurize(input_file, self.data_dir)

  def load_sparse_multitask_dataset(self):
    """Load sparse tox multitask data, sample dataset."""
    if os.path.exists(self.data_dir):
      shutil.rmtree(self.data_dir)
    featurizer = CircularFingerprint(size=1024)
    tasks = ["task1", "task2", "task3", "task4", "task5", "task6",
             "task7", "task8", "task9"]
    input_file = os.path.join(
        self.current_dir, "../../models/tests/sparse_multitask_example.csv")
    loader = DataLoader(
        tasks=tasks,
        smiles_field="smiles",
        featurizer=featurizer,
        verbosity="low")
    return loader.featurize(input_file, self.data_dir)
+1 −0

File added.

Preview size limit exceeded, changes collapsed.

+114 −11

File changed.

Preview size limit exceeded, changes collapsed.

+117 −42

File changed.

Preview size limit exceeded, changes collapsed.