(self)
| 151 | save_graphs(graph_path, self._graph) |
| 152 | |
| 153 | def load(self): |
| 154 | graph_path = os.path.join(self.save_path, "dgl_graph.bin") |
| 155 | graphs, _ = load_graphs(graph_path) |
| 156 | self._graph = graphs[0] |
| 157 | self._graph.ndata["train_mask"] = generate_mask_tensor( |
| 158 | self._graph.ndata["train_mask"].numpy() |
| 159 | ) |
| 160 | self._graph.ndata["val_mask"] = generate_mask_tensor( |
| 161 | self._graph.ndata["val_mask"].numpy() |
| 162 | ) |
| 163 | self._graph.ndata["test_mask"] = generate_mask_tensor( |
| 164 | self._graph.ndata["test_mask"].numpy() |
| 165 | ) |
| 166 | self._print_info() |
| 167 | |
| 168 | def _print_info(self): |
| 169 | if self.verbose: |
no test coverage detected