(savedir)
| 315 | |
| 316 | |
| 317 | def load_model(savedir): |
| 318 | with open('%s/word2idx.json' % savedir) as f: |
| 319 | word2idx = json.load(f) |
| 320 | npz = np.load('%s/weights.npz' % savedir) |
| 321 | W = npz['arr_0'] |
| 322 | V = npz['arr_1'] |
| 323 | return word2idx, W, V |
| 324 | |
| 325 | |
| 326 |