Merge pull request #1068 from Luckick/patch-1 (d0234844) · Commits · 钟慕尧 / deepchem

examples/sampl/sampl_graph_conv.py

+2 −2

Original line number	Diff line number	Diff line
		"""
		Script that trains graph-conv models on Tox21 dataset.
		Script that trains graph-conv models on SAMPL(FreeSolv) dataset.
		"""
		from __future__ import print_function
		from __future__ import division
		@@ -12,7 +12,7 @@ tf.set_random_seed(123)
		import deepchem as dc
		from deepchem.molnet import load_sampl

		# Load Tox21 dataset
		# Load SAMPL(FreeSolv) dataset
		SAMPL_tasks, SAMPL_datasets, transformers = load_sampl(featurizer='GraphConv')
		train_dataset, valid_dataset, test_dataset = SAMPL_datasets