Fix GraphConvTensorGraph to GraphConvModel in adme (a7a85fb0) · Commits · 钟慕尧 / deepchem

examples/adme/run_benchmarks.py

+2 −2

Original line number	Diff line number	Diff line
		@@ -14,7 +14,7 @@ from sklearn import svm
		import tensorflow as tf
		tf.set_random_seed(123)
		import deepchem as dc
		from deepchem.models.tensorgraph.models.graph_models import GraphConvTensorGraph
		from deepchem.models.tensorgraph.models.graph_models import GraphConvModel

		BATCH_SIZE = 128
		# Set to higher values to get better numbers
		@@ -74,7 +74,7 @@ def experiment(dataset_file, method='GraphConv', split='scaffold'):

		model = None
		if method == 'GraphConv':
		model = GraphConvTensorGraph(
		model = GraphConvModel(
		len(tasks), batch_size=BATCH_SIZE, mode="regression")
		elif method == 'RF':