Commit 0f374efa authored by miaecle's avatar miaecle
Browse files

yapf

parent db69e76d

deepchem/models/tensorgraph/robust_multitask.py

+1 −2
Original line number Diff line number Diff line
@@ -15,7 +15,6 @@ from deepchem.models.tensorgraph.layers import Feature, Label, Weights, \ 




class RobustMultitaskClassifier(TensorGraph):



  """Implements a neural network for robust multitasking.

  

  Key idea is to have bypass layers that feed directly from features to task

deepchem/models/tests/test_overfit.py

+0 −2
Original line number Diff line number Diff line
@@ -488,7 +488,6 @@ class TestOverfit(test_util.TensorFlowTestCase): 
    scores = model.evaluate(dataset, [regression_metric])

    assert scores[regression_metric.name] < .1



  

  def test_tf_robust_multitask_regression_overfit(self):

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

    np.random.seed(123)
@@ -526,7 +525,6 @@ class TestOverfit(test_util.TensorFlowTestCase): 
    scores = model.evaluate(dataset, [regression_metric])

    assert scores[regression_metric.name] < .2





  def test_tensorgraph_DTNN_multitask_regression_overfit(self):

    """Test deep tensor neural net overfits tiny data."""

    np.random.seed(123)

CRA Git | Maintained and supported by SUSTech CRA and CCSE