Commit 6c7dfac3 authored by Ubuntu's avatar Ubuntu
Browse files

CI

parent b3ab15b6

deepchem/models/tests/test_gcn.py

+2 −1
Original line number Diff line number Diff line
@@ -31,7 +31,8 @@ def test_gcn_regression(): 
      mode='regression',

      n_tasks=n_tasks,

      number_atom_features=30,

      batch_size=10)

      batch_size=10,

      learning_rate=0.02)



  # overfit test

  model.fit(dataset, nb_epoch=150)

CRA Git | Maintained and supported by SUSTech CRA and CCSE