Commit ec5300c8 authored by peastman's avatar peastman
Browse files

Removed debugging code

parent 7e4ae544
Loading
Loading
Loading
Loading
+0 −3
Original line number Diff line number Diff line
@@ -28,9 +28,6 @@ class TestKerasModel(unittest.TestCase):
        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])