Loading deepchem/models/tests/test_gat.py +2 −3 Original line number Diff line number Diff line Loading @@ -22,14 +22,13 @@ def test_gat_regression(): # initialize models n_tasks = len(tasks) model = GATModel( mode='regression', n_tasks=n_tasks, batch_size=4, learning_rate=0.001) model = GATModel(mode='regression', n_tasks=n_tasks, batch_size=10) # overfit test # GAT's convergence is a little slow model.fit(dataset, nb_epoch=300) scores = model.evaluate(dataset, [metric], transformers) assert scores['mean_absolute_error'] < 0.2 assert scores['mean_absolute_error'] < 0.5 @unittest.skipIf(not has_pytorch_and_pyg, Loading Loading
deepchem/models/tests/test_gat.py +2 −3 Original line number Diff line number Diff line Loading @@ -22,14 +22,13 @@ def test_gat_regression(): # initialize models n_tasks = len(tasks) model = GATModel( mode='regression', n_tasks=n_tasks, batch_size=4, learning_rate=0.001) model = GATModel(mode='regression', n_tasks=n_tasks, batch_size=10) # overfit test # GAT's convergence is a little slow model.fit(dataset, nb_epoch=300) scores = model.evaluate(dataset, [metric], transformers) assert scores['mean_absolute_error'] < 0.2 assert scores['mean_absolute_error'] < 0.5 @unittest.skipIf(not has_pytorch_and_pyg, Loading
