2016-09-04 5 views
1

私はWide and Deep learning example from Tensorflowで遊んでいます。後で予測タスクに使用する訓練されたクラシファイアを保存したいのですが、実際にどのように表示されるのかわかりません。 DNNLinearCombinedClassifierには利用可能な保存方法がなく、オブジェクトを酸洗することもできません。広いと深いテンソルフローモデルを保存

保存する方法はありますか?

答えて

0

https://www.tensorflow.org/versions/r0.10/how_tos/variables/index.html#saving-and-restoringをお探しですか?

特に変数を保存するおよび変数を復元する。さらに、チェックポイントファイルを使用すると、定期的にウェイトを保存できます(途中でトレーニングが中止された場合は、最初からやり直す必要はありません)。

+0

これは残念ながらSaverオブジェクトはモデルに公開されていないセッションにアクセスする必要があります。 –

+1

'model_dir'フラグを使ってみましたか? https://github.com/tensorflow/tensorflow/blob/master/tensorflow/contrib/learn/python/learn/estimators/dnn_linear_combined.pyによると、model_dirフラグは次のようになります: '' 'model_dir:モデルパラメータを保存するディレクトリこれはまた、ディレクトリからチェックポイントを推定値にロードして、以前に保存されたモデルを訓練するために使用することもできます.'''ので、深くて幅広いもので使用する他の分類子で使用されています。その場合は、独自の.pyファイルを作成して、すべてを復元し、必要な処理を行うことができます。 –

+0

うん、model_dirを設定すると、次回モデルをインスタンス化したときにモデルが自動的に復元されます。私はそれ以上のことをする必要があると思っていました。ありがとう! –

2

this docのチェックポイント保存セクションはあなたの質問に答える必要があります。

関連する問題