2017-02-21 10 views
0

私はテンソルフローでGoogle wide and deep modelを学習していました。テンソルフローでは、DNNLinearClassifier [wide_n_deep_tutorial.py]を使用すると、訓練された履歴モデルを復元し、ベースを訓練する方法

私のコードはtf exampleに基づいて変更されました。

私の質問は:

  1. 履歴モデルをロードし、それに基づいて新しいサンプルを訓練する方法。私はウェブ上で多くを検索しましたが、ほとんどのメソッドは次のコードのようなものです。

    セーバー

    v1 = tf.Variable(..., name="v1") 
    v2 = tf.Variable(..., name="v2") 
    ... 
    init_op = tf.initialize_all_variables() 
    saver = tf.train.Saver() 
    with tf.Session() as sess: 
        sess.run(init_op) 
        .. 
        save_path = saver.save(sess, "/tmp/model.ckpt") 
    

    の例しかしwide_n_deep電車の中で、それは返すメソッド "build_estimator" を定義:

    m = tf.contrib.learn.DNNLinearCombinedClassifierを。

    上記の方法を使用して保護者を定義すると、保存するためのVariblesがエラーになりませんでした。そして、返された "m"には、save、restoreメソッドがありません。

    model.loadtflearnで試してみましたが、動作しません。

  2. 第2は、上記と関連して、モデルを保存する方法です。あるいは、fitメソッドが/ tmp /に保存したモデルの使い方。 (チェックポイント)

私の質問は以下のように要約することができます。

私はtensorflowでDNNLinearCombinedClassifierを使用するときに、モデルを保存し、履歴モデルを復元する方法?

答えて

0

model_dirという引数があります。https://github.com/tensorflow/tensorflow/blob/master/tensorflow/contrib/learn/python/learn/estimators/dnn_linear_combined.py#L476です。モデルのチェックポイントを保存しているディレクトリを指してください。

+0

コメントありがとうございます! para model_dirがこの電車のモデルを保存するディレクトリではありませんか?または、負荷モデルのディレクトリと保存モデルのディレクトリを同じにすることはできますか?答えがイエスで、負荷が自動的に最後のモデルをロードする場合は、もう一度ありがとう。 – Cow

+0

dirは、トレーニングの前にモデルのチェックポイントをロードするため、および特定の手順の後にモデルのチェックポイントを保存するために使用できます。 – yuefengz

+0

ありがとうございました。 – Cow

関連する問題