fixes batch padding in predict: pad_batches was always "True", now it follows... (e313c5bd) · Commits · 钟慕尧 / deepchem

deepchem/models/tensorgraph/models/graph_models.py

+1 −1

Original line number	Diff line number	Diff line
		@@ -585,7 +585,7 @@ class GraphConvTensorGraph(TensorGraph):
		print('Starting epoch %i' % epoch)
		for ind, (X_b, y_b, w_b, ids_b) in enumerate(
		dataset.iterbatches(
		self.batch_size, pad_batches=True, deterministic=deterministic)):
		self.batch_size, pad_batches=pad_batches, deterministic=deterministic)):
		d = {}
		for index, label in enumerate(self.my_labels):
		if self.mode == 'classification':