c09ca92e84f45c61f9213a32c7f01627fd467174,deepchem/models/tensorgraph/tensor_graph.py,TensorGraph,predict_on_batch,#TensorGraph#Any#Any#,161

Before Change

    retval = self.predict_proba_on_batch(X, sess)
    if self.mode == "classification":
      return from_one_hot(retval, axis=2)
    return retval

  def predict_proba_on_batch(self, X, sess=None):

    def predict():

After Change

    // Returns
        A Numpy array of predictions.
    
    dataset = NumpyDataset(X=X, y=None)
    generator = self.default_generator(dataset, predict=True, pad_batches=False)
    return self.predict_on_generator(generator)

  def predict_proba_on_batch(self, X, sess=None):
    dataset = NumpyDataset(X=X, y=None)
    generator = self.default_generator(dataset, predict=True, pad_batches=False)

Instances