Loading deepchem/feat/graph_data.py +1 −1 Original line number Diff line number Diff line Loading @@ -123,7 +123,7 @@ class GraphData: edge_attr=edge_features, pos=node_pos_features) def to_dgl_graph(self, self_loop=False): def to_dgl_graph(self, self_loop: bool=False): """Convert to DGL graph data instance Returns Loading deepchem/models/tests/test_gcn.py +1 −1 Original line number Diff line number Diff line Loading @@ -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=50) model.fit(dataset, nb_epoch=60) scores = model.evaluate(dataset, [metric], transformers) assert scores['mean_absolute_error'] < 0.5 Loading Loading
deepchem/feat/graph_data.py +1 −1 Original line number Diff line number Diff line Loading @@ -123,7 +123,7 @@ class GraphData: edge_attr=edge_features, pos=node_pos_features) def to_dgl_graph(self, self_loop=False): def to_dgl_graph(self, self_loop: bool=False): """Convert to DGL graph data instance Returns Loading
deepchem/models/tests/test_gcn.py +1 −1 Original line number Diff line number Diff line Loading @@ -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=50) model.fit(dataset, nb_epoch=60) scores = model.evaluate(dataset, [metric], transformers) assert scores['mean_absolute_error'] < 0.5 Loading
