2017-05-19 6 views
0

私はKerasを使ってテキストクラシファイアをトレーニングすることができ、モデルも保存できます。私の質問は、未知のテキストを予測するためのモデルをロードするときです。トレーニングプロセス中にエンコードされたのと同じ方法で入力テキストをエンコードする必要があります。トレーニング部分でマッピングを保存し、予測を行うときにどのようにリロードすることができますか?テキストインデックスのエンコードマッピングを保存します

これは、単語をトレーニングのインデックスにマップするために使用しているコードです。テンソルフローからVocabularyProcessorを使用します。

processor = learn.preprocessing.VocabularyProcessor(1000) 
x = np.array(list(processor.fit_transform(x_raw))) 
y = np.array(y_raw) 

ありがとうございます!

答えて

0

考え出し:

processor.save(...) 

learn.preprocessing.VocabularyProcessor.restore(...)