Commit 3679181d authored by miaecle's avatar miaecle
Browse files

yapf

parent f1244991

deepchem/models/tensorgraph/progressive_multitask.py

+9 −9
Original line number Diff line number Diff line
@@ -153,8 +153,7 @@ class ProgressiveMultitaskRegressor(TensorGraph): 
      self.add_output(output_layer)



      task_loss = self.create_loss(layer)

      self.create_submodel(

          layers=task_layers, loss=task_loss, optimizer=None)

      self.create_submodel(layers=task_layers, loss=task_loss, optimizer=None)

    # Weight decay not activated

    """

    if weight_decay_penalty != 0.0:
@@ -293,6 +292,7 @@ class ProgressiveMultitaskRegressor(TensorGraph): 
    out = np.stack(retval, axis=1)

    return out





class ProgressiveMultitaskClassifier(ProgressiveMultitaskRegressor):

  """Implements a progressive multitask neural network for classification.

deepchem/models/tests/test_overfit.py

+1 −3
Original line number Diff line number Diff line
@@ -885,8 +885,7 @@ class TestOverfit(test_util.TensorFlowTestCase): 


    dataset = dc.data.NumpyDataset(X, y, w, ids)



    metric = dc.metrics.Metric(

        dc.metrics.accuracy_score, task_averager=np.mean)

    metric = dc.metrics.Metric(dc.metrics.accuracy_score, task_averager=np.mean)

    model = dc.models.ProgressiveMultitaskClassifier(

        n_tasks,

        n_features,
@@ -906,7 +905,6 @@ class TestOverfit(test_util.TensorFlowTestCase): 
    scores = model.evaluate(dataset, [metric])

    assert scores[metric.name] > .9





  def test_tf_progressive_regression_overfit(self):

    """Test tf progressive multitask overfits tiny data."""

    np.random.seed(123)

CRA Git | Maintained and supported by SUSTech CRA and CCSE