def persist(self, model_dir: Text) -> Optional[Dict[Text, Any]]:
"""Persist this model into the passed directory.
Return the metadata necessary to load the model again."""
regex_file = os.path.join(model_dir, REGEX_FEATURIZER_FILE_NAME)
utils.write_json_to_file(regex_file, self.known_patterns, indent=4)
return {"regex_file": REGEX_FEATURIZER_FILE_NAME}
def persist(self, model_dir):
# type: (Text) -> Optional[Dict[Text, Any]]
if self.synonyms:
entity_synonyms_file = os.path.join(model_dir,
ENTITY_SYNONYMS_FILE_NAME)
write_json_to_file(entity_synonyms_file, self.synonyms,
separators=(',', ': '))
return {"synonyms_file": ENTITY_SYNONYMS_FILE_NAME}
def persist(self, model_dir):
# type: (Text) -> None
"""Persists the metadata of a model to a given directory."""
metadata = self.metadata.copy()
metadata.update({
"trained_at":
datetime.datetime.now().strftime('%Y%m%d-%H%M%S'),
"rasa_nlu_version":
rasa_nlu_gao.__version__,
})
filename = os.path.join(model_dir, 'metadata.json')
write_json_to_file(filename, metadata, indent=4)