save the graph list and the labels
(self)
| 421 | return False |
| 422 | |
| 423 | def save(self): |
| 424 | """save the graph list and the labels""" |
| 425 | graph_path = os.path.join(self.save_path, self.save_name + ".bin") |
| 426 | info_path = os.path.join(self.save_path, self.save_name + ".pkl") |
| 427 | save_graphs(str(graph_path), self._hg) |
| 428 | save_info( |
| 429 | str(info_path), |
| 430 | { |
| 431 | "num_classes": self.num_classes, |
| 432 | "predict_category": self.predict_category, |
| 433 | }, |
| 434 | ) |
| 435 | |
| 436 | def load(self): |
| 437 | """load the graph list and the labels from disk""" |
nothing calls this directly
no test coverage detected