Loading deepchem/models/tensorgraph/tensor_graph.py +0 −1 Original line number Original line Diff line number Diff line Loading @@ -257,7 +257,6 @@ class TensorGraph(Model): result = np.array(sess.run(out_tensors, feed_dict=feed_dict)) result = np.array(sess.run(out_tensors, feed_dict=feed_dict)) if len(result.shape) == 3: if len(result.shape) == 3: result = np.transpose(result, axes=[1, 0, 2]) result = np.transpose(result, axes=[1, 0, 2]) result = result.squeeze(axis=0) result = undo_transforms(result, transformers) result = undo_transforms(result, transformers) results.append(result) results.append(result) return np.concatenate(results, axis=0) return np.concatenate(results, axis=0) Loading Loading
deepchem/models/tensorgraph/tensor_graph.py +0 −1 Original line number Original line Diff line number Diff line Loading @@ -257,7 +257,6 @@ class TensorGraph(Model): result = np.array(sess.run(out_tensors, feed_dict=feed_dict)) result = np.array(sess.run(out_tensors, feed_dict=feed_dict)) if len(result.shape) == 3: if len(result.shape) == 3: result = np.transpose(result, axes=[1, 0, 2]) result = np.transpose(result, axes=[1, 0, 2]) result = result.squeeze(axis=0) result = undo_transforms(result, transformers) result = undo_transforms(result, transformers) results.append(result) results.append(result) return np.concatenate(results, axis=0) return np.concatenate(results, axis=0) Loading
