(self)
| 458 | return os.path.join(self.save_path, "tu_{}.pkl".format(self.name)) |
| 459 | |
| 460 | def save(self): |
| 461 | label_dict = {"labels": self.graph_labels} |
| 462 | info_dict = { |
| 463 | "max_num_node": self.max_num_node, |
| 464 | "num_labels": self.num_labels, |
| 465 | } |
| 466 | save_graphs(str(self.graph_path), self.graph_lists, label_dict) |
| 467 | save_info(str(self.info_path), info_dict) |
| 468 | |
| 469 | def load(self): |
| 470 | graphs, label_dict = load_graphs(str(self.graph_path)) |