Loading deepchem/models/tensorgraph/tensor_graph.py +2 −2 Original line number Diff line number Diff line Loading @@ -301,8 +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: result = undo_transforms(np.concatenate(feed_results, 1), transformers) feed_results = [result[:, i:i+1] for i in range(result.shape[1])] 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 +2 −2 Original line number Diff line number Diff line Loading @@ -301,8 +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: result = undo_transforms(np.concatenate(feed_results, 1), transformers) feed_results = [result[:, i:i+1] for i in range(result.shape[1])] 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
