def save(self, fname=None):
"""Save the parameters of the agent to a file."""
fname = self.opt.get('model_file', None) if fname is None else fname
if fname:
if self.model_type == 'nn':
print("[ saving model: " + fname + " ]")
self.model.save_weights(fname + '.h5')
self.embedding_dict.save_items(fname)
if self.model_type == 'ngrams':
print("[ saving model: " + fname + " ]")
with open(fname + '_cls.pkl', 'wb') as model_file:
pickle.dump(self.model, model_file)
with open(fname + '_opt.json', 'w') as opt_file:
json.dump(self.opt, opt_file)
评论列表
文章目录