Commit cc7e7f45 authored by mufeili's avatar mufeili
Browse files

CI

parent ac3c317b

deepchem/models/tests/test_gat.py

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



  # overfit test

  model.fit(dataset, nb_epoch=70)

  model.fit(dataset, nb_epoch=80)

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

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

deepchem/models/tests/test_gcn.py

+2 −2
Original line number Diff line number Diff line
@@ -79,9 +79,9 @@ def test_gcn_reload(): 
      number_atom_features=30,

      model_dir=model_dir,

      batch_size=10,

      learning_rate=0.001)

      learning_rate=0.0003)



  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

CRA Git | Maintained and supported by SUSTech CRA and CCSE