Trying to debug failure that only happens on travis

        learning_rate=0.005)

    model.fit(dataset, nb_epoch=1000)

    prediction = np.squeeze(model.predict_on_batch(X))

    print(y)

    print(prediction)

    print(np.round(prediction))

    assert np.array_equal(y, np.round(prediction))

    metric = dc.metrics.Metric(dc.metrics.roc_auc_score)

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