コード例 #1
0
ファイル: tf_io.py プロジェクト: wayne9qiu/derek 
def save_classifier(path, extractor, feature_computer, graph, session, saver):
    extractor_path = os.path.join(path, 'extractor')
    os.makedirs(extractor_path, exist_ok=True)
    extractor.save(extractor_path)

    feature_computer_path = os.path.join(path, 'feature_computer')
    os.makedirs(feature_computer_path, exist_ok=True)
    feature_computer.save(feature_computer_path)

    save_tf_model_with_saver(path, 'model', session, saver)
    save_with_pickle(_convert_save_graph(graph), path, "graph.pkl")
コード例 #2
0
 def save(self, out_path):
     save_classifier(out_path, self.extractor, self.feature_computer, self.graph, self.session, self.saver)
     save_with_pickle(self.collapser, out_path, "collapser")
コード例 #3
0
 def save(self, out_path):
     save_with_pickle(self, out_path, "feature_extractor.pkl")
コード例 #4
0
ファイル: classifier.py プロジェクト: ispras-texterra/derek 
 def save(self, out_path):
     save_classifier(out_path, self.extractor, self.feature_computer,
                     self.graph, self.session, self.saver)
     save_with_pickle(self.post_processor, out_path, "post_processor")
コード例 #5
0
ファイル: manager.py プロジェクト: wayne9qiu/derek 
 def save(self, out_path):
     save_with_pickle(self, out_path, "feature_computer.pkl")
コード例 #6
0
ファイル: readers.py プロジェクト: wayne9qiu/derek 
 def dump_docs(path: str, docs: List[Document], name: str):
     save_with_pickle(docs, path, name)
コード例 #7
0
ファイル: readers.py プロジェクト: wayne9qiu/derek 
def dump(path: str, docs: list):
    save_with_pickle(docs, path, "docs.pkl")