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

Attempted test_api fix. Turning off dropout.

parent 44b3d86c
Loading
Loading
Loading
Loading
+50 −51
Original line number Diff line number Diff line
@@ -282,8 +282,6 @@ class TestModelAPI(TestAPI):
  #                        Metric(metrics.mean_squared_error),
  #                        Metric(metrics.mean_absolute_error)]

  #  model = SingleTaskDNN(tasks, task_types, model_params, self.model_dir)

  #  # Fit trained model
  #  model.fit(train_dataset)
  #  model.save()
@@ -301,15 +299,16 @@ class TestModelAPI(TestAPI):
    """Straightforward test of Keras multitask deepchem classification API."""
    ############################################################# DEBUG
    g = tf.Graph()
    #sess = tf.Session(graph=g)
    sess = tf.Session()
    sess = tf.Session(graph=g)
    K.set_session(sess)
    with g.as_default():
    ############################################################# DEBUG
      task_type = "classification"
      # TODO(rbharath): There should be some automatic check to ensure that all
      # required model_params are specified.
      # TODO(rbharath): Turning off dropout to make tests behave.
      model_params = {"nb_hidden": 10, "activation": "relu",
                    "dropout": .5, "learning_rate": .01,
                      "dropout": .0, "learning_rate": .01,
                      "momentum": .9, "nesterov": False,
                      "decay": 1e-4, "batch_size": 5,
                      "nb_epoch": 2, "init": "glorot_uniform",