2016-07-24 5 views
0

내 텍스트 코퍼스 용 Word2Vec 모델을 교육하고 싶습니다. TensorFlow의 튜토리얼에서 코드를 얻을 수 있습니다. 내가 모르는 것은 CNN 텍스트 분류에 사용하기 위해이 모델을 저장하는 방법입니다. 피클을 사용하여 저장 한 다음 나중에 읽어야합니까?CNN 텍스트 분류 용 Word2Vec 저장

답변

0

pickle을 사용하여 디스크에 저장할 수 있습니다. 그런 다음 CNN 모델을 만들 때 저장된 단어 포함 테이블을로드하고이를 사용하여 CNN 분류 자에 대한 단어 삽입을 보유하는 TensorFlow 변수를 초기화합니다.

0

텐롤 흐름의 경우 모델을 저장하는 방식이 산세 처리가 아닙니다.

Tensorflow는 모델을 내보내기하기위한 프로토 bufs로 모델을 저장하는 데 사용되는 텐서 흐름을 제공합니다. 모델을 저장하는 방법은 tensorflow 세션을 저장하는 것 같은 : saver.save (SESS, 'my_test_model', global_step = 1000)을 Heres

완전한 답변에 대한 링크 : Tensorflow: how to save/restore a model?