def persist(self, file_name: Text, model_dir: Text) -> Optional[Dict[Text, Any]]: """Persist this component to disk for future loading.""" file_name = file_name + ".pkl" classifier_file = os.path.join(model_dir, file_name) utils.pycloud_pickle(classifier_file, self) return {"file": file_name}
def persist(self, file_name: Text, model_dir: Text) -> Optional[Dict[Text, Any]]: """Persist this model into the passed directory.""" file_name = file_name + ".pkl" classifier_file = os.path.join(model_dir, file_name) utils.pycloud_pickle(classifier_file, self) return {"file": file_name}
def persist(self, file_name: Text, model_dir: Text) -> Optional[Dict[Text, Any]]: """Persist this model into the passed directory. Returns the metadata necessary to load the model again. """ file_name = file_name + ".pkl" featurizer_file = os.path.join(model_dir, file_name) utils.pycloud_pickle(featurizer_file, self) return {"file": file_name}
def persist(self, file_name: Text, model_dir: Text) -> Optional[Dict[Text, Any]]: """Persist this component to disk for future loading.""" # n2g model just for function test n2g_file = os.path.join(model_dir, 'n2g_params.pkl') self.n2g.save(n2g_file) # total object file_name = file_name + ".pkl" classifier_file = os.path.join(model_dir, file_name) utils.pycloud_pickle(classifier_file, self) return {"file": file_name}
def persist(self, file_name: Text, model_dir: Text) -> Optional[Dict[Text, Any]]: """Persist this model into the passed directory.""" classifier_file_name = file_name + "_classifier.pkl" encoder_file_name = file_name + "_encoder.pkl" if self.clf: utils.pycloud_pickle(os.path.join(model_dir, classifier_file_name), self.clf) utils.pycloud_pickle(os.path.join(model_dir, encoder_file_name), self.le) return { "classifier": classifier_file_name, "encoder": encoder_file_name }