(self)
| 242 | save_info(emb_path, {"embed": self.pretrained_emb}) |
| 243 | |
| 244 | def load(self): |
| 245 | emb_path = os.path.join(self.save_path, "emb.pkl") |
| 246 | |
| 247 | self._trees = load_graphs(self.graph_path)[0] |
| 248 | self._vocab = load_info(self.vocab_path)["vocab"] |
| 249 | self._pretrained_emb = None |
| 250 | if os.path.exists(emb_path): |
| 251 | self._pretrained_emb = load_info(emb_path)["embed"] |
| 252 | |
| 253 | @property |
| 254 | def vocab(self): |
nothing calls this directly
no test coverage detected