Commit cd9a5536 authored by mufeili's avatar mufeili
Browse files

Update

parent 9bc97f2a

deepchem/models/tests/test_gat.py

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



  # overfit test

  model.fit(dataset, nb_epoch=400)

  model.fit(dataset, nb_epoch=500)

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

  assert scores['mean_absolute_error'] < 0.5

CRA Git | Maintained and supported by SUSTech CRA and CCSE