Commit 9d3fca5d authored by mufeili's avatar mufeili
Browse files

Update

parent 22e5c756

deepchem/models/tests/test_gcn.py

+1 −1
Original line number Diff line number Diff line
@@ -29,7 +29,7 @@ def test_gcn_regression(): 
    model = GCNModel(mode='regression', n_tasks=n_tasks, number_atom_features=30, batch_size=10)



    # overfit test

    model.fit(dataset, nb_epoch=60)

    model.fit(dataset, nb_epoch=100)

    scores = model.evaluate(dataset, [metric], transformers)

    assert scores['mean_absolute_error'] < 0.5

CRA Git | Maintained and supported by SUSTech CRA and CCSE