Commit 869b8ca8 authored by mufeili's avatar mufeili
Browse files

Fix

parent aee8c676

deepchem/models/tests/test_attentivefp.py

+1 −1
Original line number Diff line number Diff line
@@ -52,7 +52,7 @@ def test_attentivefp_classification(): 
      learning_rate=0.001)



  # overfit test

  model.fit(dataset, nb_epoch=60)

  model.fit(dataset, nb_epoch=70)

  scores = model.evaluate(dataset, [metric], transformers)

  assert scores['mean-roc_auc_score'] >= 0.85

deepchem/models/tests/test_gcn.py

+1 −1
Original line number Diff line number Diff line
@@ -34,7 +34,7 @@ def test_gcn_regression(): 
      batch_size=10)



  # overfit test

  model.fit(dataset, nb_epoch=110)

  model.fit(dataset, nb_epoch=150)

  scores = model.evaluate(dataset, [metric], transformers)

  assert scores['mean_absolute_error'] < 0.5

CRA Git | Maintained and supported by SUSTech CRA and CCSE