Commit ac3c317b authored by mufeili's avatar mufeili
Browse files

Fix

parent 3a57d32c

deepchem/models/tests/test_gcn.py

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

      learning_rate=0.001)



  model.fit(dataset, nb_epoch=50)

  model.fit(dataset, nb_epoch=60)

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

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

deepchem/models/tests/test_graph_models.py

+1 −1
Original line number Diff line number Diff line
@@ -50,7 +50,7 @@ def test_graph_conv_model(): 
      batch_normalize=False,

      mode='classification')



  model.fit(dataset, nb_epoch=10)

  model.fit(dataset, nb_epoch=20)

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

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

CRA Git | Maintained and supported by SUSTech CRA and CCSE