Commit ad0df487 authored by Bharath Ramsundar's avatar Bharath Ramsundar
Browse files

Local changes

parent 49544f33

deepchem/datasets/__init__.py

+2 −1
Original line number Diff line number Diff line
@@ -259,7 +259,8 @@ class NumpyDataset(Dataset): 
    """Get the weight vector for this dataset as a single numpy array."""

    return self._w



  def iterbatches(self, batch_size=None, epoch=0, deterministic=False, pad_batches=False):

  def iterbatches(self, batch_size=None, epoch=0, deterministic=False,

                  pad_batches=False):

    """Get an object that iterates over minibatches from the dataset.



    Each minibatch is returned as a tuple of four numpy arrays: (X, y, w, ids).

deepchem/models/multitask.py

+1 −1
Original line number Diff line number Diff line
@@ -103,7 +103,7 @@ class SingletaskToMultitask(Model): 
      log("Fitting model for task %s" % task, self.verbosity, "high")

      task_model = self.model_builder(

          self.task_model_dirs[task])

      task_model.fit(task_datasets[ind])

      task_model.fit(task_datasets[ind], **kwargs)

      task_model.save()



  def predict_on_batch(self, X):

deepchem/models/tensorflow_models/__init__.py

+2 −1
Original line number Diff line number Diff line
@@ -707,7 +707,8 @@ class TensorflowModel(Model): 
    Returns:

      y_pred: numpy ndarray of shape (n_samples,)

    """

    return Model.predict(self, dataset, transformers, self.model_instance.batch_size, True)

    return Model.predict(self, dataset, transformers,

                         self.model_instance.batch_size, True)



  def predict_on_batch(self, X, pad_batch=True):

    """

CRA Git | Maintained and supported by SUSTech CRA and CCSE