Loading deepchem/models/tests/test_pagtn.py +1 −1 Original line number Diff line number Diff line Loading @@ -30,7 +30,7 @@ def test_pagtn_regression(): model = PagtnModel(mode='regression', n_tasks=n_tasks, batch_size=16) # overfit test model.fit(dataset, nb_epoch=20) model.fit(dataset, nb_epoch=100) scores = model.evaluate(dataset, [metric], transformers) assert scores['mean_absolute_error'] < 0.5 Loading Loading
deepchem/models/tests/test_pagtn.py +1 −1 Original line number Diff line number Diff line Loading @@ -30,7 +30,7 @@ def test_pagtn_regression(): model = PagtnModel(mode='regression', n_tasks=n_tasks, batch_size=16) # overfit test model.fit(dataset, nb_epoch=20) model.fit(dataset, nb_epoch=100) scores = model.evaluate(dataset, [metric], transformers) assert scores['mean_absolute_error'] < 0.5 Loading
