2016-12-15 18 views
0

私はちょうどtensorflow 0.12で、次の警告を削除するtf.contrib.learn tutorialにコードをリファクタリングするための好ましい方法は何疑問に思って:tf.contrib.learnチュートリアル廃止の警告

を見て
Estimator is decoupled from Scikit Learn interface by moving into 
separate class SKCompat. Arguments x, y and batch_size are only 
available in the SKCompat class, Estimator will only accept input_fn. 
Example conversion: 
est = Estimator(...) -> est = SKCompat(Estimator(...)) 

コード、DNNClassifierは、カスタム見積もりを取ることができない、とフィット法は、x、y、およびBATCH_SIZEがフィットメソッドに渡されていない場合でも、警告をトリガー

self._estimator.fit(x=x, 
        y=y, 
        input_fn=input_fn, 
        steps=steps, 
        batch_size=batch_size, 
        monitors=hooks, 
        max_steps=max_steps) 

を呼び出します。

答えて

0

Tensorflow 1.0(まもなくリリース予定)の時点で、それはtf.learnからtf.contrib.learnに変更されます。あなたはそれを試してみたいと思うなら、それはうまくいくかもしれません。

+0

で説明したようにinput_fn()を使用することもできます。公式サポートされている高レベルライブラリがあればいいと思います。 – ComputerScientist

1

SKCompatで「ラップ」しようとするとよいでしょう。 私は偶然の問題を抱えていましたが、私は部分的に(例)tf.contrib.learn.SKCompat(tf.contrib.learn.DNNClassifier(...).fit(...)で解決しました。 tensorflow.org

関連する問題