Commit 3d14eb58 authored by peastman's avatar peastman
Browse files

Increased number of epochs on test

parent 4698cad4

deepchem/models/tensorgraph/models/test_graph_models.py

+1 −1
Original line number Diff line number Diff line
@@ -374,7 +374,7 @@ class TestGraphModels(unittest.TestCase): 
        mode="regression")



    # Fit trained model

    model.fit(dataset, nb_epoch=200)

    model.fit(dataset, nb_epoch=250)



    # Eval model on train

    pred = model.predict(dataset)

CRA Git | Maintained and supported by SUSTech CRA and CCSE