Fix GraphConvTensorGraph to GraphConvModel in chembl example (48672c67) · Commits · 钟慕尧 / deepchem

examples/chembl/chembl_graph_conv.py

+2 −2

Original line number	Diff line number	Diff line
		@@ -7,7 +7,7 @@ from __future__ import unicode_literals

		import numpy as np

		from models import GraphConvTensorGraph
		from models import GraphConvModel

		np.random.seed(123)
		import tensorflow as tf
		@@ -30,7 +30,7 @@ n_feat = 75
		# Batch size of models
		batch_size = 128

		model = GraphConvTensorGraph(
		model = GraphConvModel(
		len(chembl_tasks), batch_size=batch_size, mode='regression')

		# Fit trained model