Loading deepchem/models/tests/test_gat.py +2 −1 Original line number Diff line number Diff line Loading @@ -32,7 +32,7 @@ def test_gat_regression(): # 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.5 assert scores['mean_absolute_error'] < 0.75 @unittest.skipIf(not has_pytorch_and_pyg, Loading Loading @@ -78,6 +78,7 @@ def test_gat_reload(): model.fit(dataset, nb_epoch=150) scores = model.evaluate(dataset, [metric], transformers) assert scores['mean-roc_auc_score'] >= 0.70 reloaded_model = GATModel( mode='classification', Loading Loading
deepchem/models/tests/test_gat.py +2 −1 Original line number Diff line number Diff line Loading @@ -32,7 +32,7 @@ def test_gat_regression(): # 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.5 assert scores['mean_absolute_error'] < 0.75 @unittest.skipIf(not has_pytorch_and_pyg, Loading Loading @@ -78,6 +78,7 @@ def test_gat_reload(): model.fit(dataset, nb_epoch=150) scores = model.evaluate(dataset, [metric], transformers) assert scores['mean-roc_auc_score'] >= 0.70 reloaded_model = GATModel( mode='classification', Loading
