Commit 57424564 authored by peastman's avatar peastman
Browse files

Trying to get test to pass on travis

parent 06697c18

deepchem/models/tests/test_kerasmodel.py

+1 −1
Original line number Diff line number Diff line
@@ -24,7 +24,7 @@ class TestKerasModel(unittest.TestCase): 
        keras_model,

        dc.models.losses.SigmoidCrossEntropy(),

        output_types=['prediction', 'loss'],

        learning_rate=0.005)

        learning_rate=0.004)

    model.fit(dataset, nb_epoch=1000)

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

    print(y)

CRA Git | Maintained and supported by SUSTech CRA and CCSE