Commit b9073cc3 authored by mufeili's avatar mufeili
Browse files

Format

parent af9644c7
Loading
Loading
Loading
Loading
+1 −2
Original line number Diff line number Diff line
@@ -32,8 +32,7 @@ def test_gat_regression():
      n_tasks=n_tasks,
      number_atom_features=30,
      batch_size=10,
      learning_rate=0.001
  )
      learning_rate=0.001)

  # overfit test
  model.fit(dataset, nb_epoch=300)