Esempio n. 1
0
    def persist(self, model_dir):
        # type: (Text) -> Optional[Dict[Text, Any]]
        """Persist this model into the passed directory."""

        classifier_file = os.path.join(model_dir, NGRAM_MODEL_FILE_NAME)
        utils.pycloud_pickle(classifier_file, self)
        return {"classifier_file": NGRAM_MODEL_FILE_NAME}
    def persist(self, model_dir):
        # type: (Text) -> Optional[Dict[Text, Any]]
        """Persist this model into the passed directory."""

        classifier_file = os.path.join(model_dir, SKLEARN_MODEL_FILE_NAME)
        utils.pycloud_pickle(classifier_file, self)
        return {"classifier_file": SKLEARN_MODEL_FILE_NAME}
 def persist(self, model_dir):
     # type: (Text) -> Dict[Text, Any]
     """Persist this model into the passed directory.
     Returns the metadata necessary to load the model again."""
     featurizer_file = os.path.join(model_dir, self.name + ".pkl")
     utils.pycloud_pickle(featurizer_file, self)
     return {"tfidf_sentence_embedding_file": self.name + ".pkl"}
Esempio n. 4
0
    def persist(self, model_dir):
        # type: (Text) -> Optional[Dict[Text, Any]]
        """Persist this model into the passed directory."""

        featurizer_file = os.path.join(model_dir, NGRAM_MODEL_FILE_NAME)
        utils.pycloud_pickle(featurizer_file, self)
        return {"featurizer_file": NGRAM_MODEL_FILE_NAME}
    def persist(self, model_dir):
        # type: (Text) -> Dict[Text, Any]
        """Persist this model into the passed directory.
        Returns the metadata necessary to load the model again."""

        featurizer_file = os.path.join(model_dir, self.name + ".pkl")
        utils.pycloud_pickle(featurizer_file, self)
        return {"featurizer_file": self.name + ".pkl"}
Esempio n. 6
0
    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."""

        file_name = file_name + ".pkl"
        classifier_file = os.path.join(model_dir, file_name)
        utils.pycloud_pickle(classifier_file, self)
        return {"file": file_name}
Esempio n. 8
0
    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}
Esempio n. 9
0
    def persist(self, model_dir: Text) -> Optional[Dict[Text, Any]]:
        """Persist this model into the passed directory."""

        featurizer_file = os.path.join(model_dir, NGRAM_MODEL_FILE_NAME)
        utils.pycloud_pickle(featurizer_file, self)
        return {"featurizer_file": NGRAM_MODEL_FILE_NAME}