Commit 42a53cdc authored by peastman's avatar peastman
Browse files

Debugging

parent 44a932cc
Loading
Loading
Loading
Loading
+3 −0
Original line number Diff line number Diff line
@@ -27,6 +27,9 @@ 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.isclose(prediction, y.flatten(), atol=0.4))
    assert np.all(np.isclose(prediction, y.flatten(), atol=0.4))
    metric = dc.metrics.Metric(dc.metrics.roc_auc_score)
    scores = model.evaluate(dataset, [metric])