Commit aee8c676 authored by mufeili's avatar mufeili
Browse files

Update

parent cc7e7f45
Loading
Loading
Loading
Loading
+1 −1
Original line number Diff line number Diff line
@@ -34,7 +34,7 @@ def test_gat_regression():
      batch_size=10)

  # overfit test
  model.fit(dataset, nb_epoch=100)
  model.fit(dataset, nb_epoch=150)
  scores = model.evaluate(dataset, [metric], transformers)
  assert scores['mean_absolute_error'] < 0.5