Commit 16e8ae88 authored by miaecle's avatar miaecle
Browse files

fix test failure

parent 145ba9a1

deepchem/models/tests/test_overfit.py

+1 −1
Original line number Original line Diff line number Diff line
@@ -1005,7 +1005,7 @@ class TestOverfit(test_util.TensorFlowTestCase): 
        mode="regression")

        mode="regression")





    # Fit trained model

    # Fit trained model

    model.fit(dataset, nb_epoch=40)

    model.fit(dataset, nb_epoch=100)





    # Eval model on train

    # Eval model on train

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

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

CRA Git | Maintained and supported by SUSTech CRA and CCSE