Commit 51f7ead6 authored by Bharath Ramsundar's avatar Bharath Ramsundar
Browse files

Some more test fixes

parent 633e03f1
Loading
Loading
Loading
Loading
+1 −1
Original line number Diff line number Diff line
@@ -23,7 +23,7 @@ class TestTFHyperparamOptAPI(TestAPI):
  """
  Test hyperparameter optimization API.
  """
  def test_multitask_keras_mlp_ECFP_classification_hyperparam_opt(self):
  def test_multitask_tf_mlp_ECFP_classification_hyperparam_opt(self):
    """Straightforward test of Tensorflow multitask deepchem classification API."""
    splittype = "scaffold"
    output_transformers = []
+7 −2
Original line number Diff line number Diff line
@@ -151,7 +151,8 @@ class Model(object):
    #y_pred = undo_transforms(y_pred, transformers)

    batch_size = self.model_params["batch_size"]
    y_preds = []
    # Have to include ys/ws since we might pad batches
    ys, w_preds, y_preds = [], [], []
    print("predict()")
    print("len(dataset)")
    print(len(dataset))
@@ -159,9 +160,13 @@ class Model(object):
      y_pred_batch = np.reshape(self.predict_on_batch(X_batch), y_batch.shape)
      y_pred_batch = undo_transforms(y_pred_batch, transformers)
      y_preds.append(y_pred_batch)
      ys.append(y_batch)
      w_preds.append(w_batch)
      print("X_batch.shape, y_batch.shape, y_pred_batch.shape")
      print(X_batch.shape, y_batch.shape, y_pred_batch.shape)
    y = np.vstack(ys)
    y_pred = np.vstack(y_preds)
    w_pred = np.vstack(w_preds)
  
    #X = X[w.flatten() != 0, :]
    #print("Model.predict()")
@@ -173,7 +178,7 @@ class Model(object):
    #  y_task = to_one_hot(y_task)
    #  y_pred_task = y_pred_task[w_task.flatten() != 0][:, np.newaxis]

    return y_pred
    return y, y_pred, w_pred

    #task_names = dataset.get_task_names()
    #pred_task_names = ["%s_pred" % task_name for task_name in task_names]
+4 −4
Original line number Diff line number Diff line
@@ -66,9 +66,9 @@ class Evaluator(object):
    Computes statistics of model on test data and saves results to csv.
    """
    #pred_y_df = self.model.predict(self.dataset, self.transformers)
    y = self.dataset.get_labels()
    w = self.dataset.get_weights()
    y_pred = self.model.predict(self.dataset, self.transformers)
    #y = self.dataset.get_labels()
    #w = self.dataset.get_weights()
    y, y_pred, w_pred = self.model.predict(self.dataset, self.transformers)
    multitask_scores = {}

    colnames = ["task_name"] + [metric.name for metric in metrics]
@@ -111,7 +111,7 @@ class Evaluator(object):

    # Compute multitask metrics
    for metric in metrics:
      multitask_scores[metric.name] = metric.compute_metric(y, y_pred, w)
      multitask_scores[metric.name] = metric.compute_metric(y, y_pred, w_pred)
    print("multitask_scores")
    print(multitask_scores)