fix bugs (ce747f98) · Commits · 钟慕尧 / deepchem

deepchem/models/tensorgraph/models/graph_models.py

+12 −2

Original line number	Diff line number	Diff line
		@@ -187,6 +187,11 @@ class WeaveTensorGraph(TensorGraph):
		generator,
		transformers=transformers,
		outputs=outputs)
		if outputs == None:
		outputs = self.outputs
		if len(outputs) == 1:
		return outputs
		else:
		return np.stack(outputs, axis=1)

		class DTNNTensorGraph(TensorGraph):
		@@ -508,6 +513,11 @@ class DAGTensorGraph(TensorGraph):
		generator,
		transformers=transformers,
		outputs=outputs)
		if outputs == None:
		outputs = self.outputs
		if len(outputs) == 1:
		return outputs
		else:
		return np.stack(outputs, axis=1)

		class PetroskiSuchTensorGraph(TensorGraph):

+6 −1

Original line number	Diff line number	Diff line
		@@ -276,4 +276,9 @@ class TextCNNTensorGraph(TensorGraph):
		generator,
		transformers=transformers,
		outputs=outputs)
		if outputs == None:
		outputs = self.outputs
		if len(outputs) == 1:
		return outputs
		else:
		return np.stack(outputs, axis=1)
		No newline at end of file