Advertisement
Guest User

Untitled

a guest
Feb 10th, 2016
65
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.54 KB | None | 0 0
  1. def save_file(filepath):
  2. f = h5py.File(filepath, 'w')
  3. f.attrs['nb_documents'] = len(self.layers)
  4. for k, l in enumerate(self.layers):
  5. g = f.create_group('doc_{}'.format(k))
  6. weights = l.get_weights()
  7. g.attrs['nb_params'] = len(weights)
  8. for n, param in enumerate(weights):
  9. param_name = 'param_{}'.format(n)
  10. param_dset = g.create_dataset(param_name, param.shape,
  11. dtype=param.dtype)
  12. param_dset[:] = param
  13. f.flush()
  14. f.close()
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement