Loading deepchem/models/tensorgraph/tensor_graph.py +8 −9 Original line number Diff line number Diff line Loading @@ -301,9 +301,8 @@ class TensorGraph(Model): feed_dict[self._training_placeholder] = 0.0 feed_results = self.session.run(outputs, feed_dict=feed_dict) if len(feed_results) > 1: if len(transformers): raise ValueError("Does not support transformations " "for multiple outputs.") result = undo_transforms(np.stack(feed_results, 1), transformers) feed_results = [result[:, i] for i in range(result.shape[1])] elif len(feed_results) == 1: result = undo_transforms(feed_results[0], transformers) feed_results = [result] Loading Loading
deepchem/models/tensorgraph/tensor_graph.py +8 −9 Original line number Diff line number Diff line Loading @@ -301,9 +301,8 @@ class TensorGraph(Model): feed_dict[self._training_placeholder] = 0.0 feed_results = self.session.run(outputs, feed_dict=feed_dict) if len(feed_results) > 1: if len(transformers): raise ValueError("Does not support transformations " "for multiple outputs.") result = undo_transforms(np.stack(feed_results, 1), transformers) feed_results = [result[:, i] for i in range(result.shape[1])] elif len(feed_results) == 1: result = undo_transforms(feed_results[0], transformers) feed_results = [result] Loading
