tf.contrib.learn教程弃用警告

时间:2016-12-15 23:29:17

标签: python tensorflow

我只是想知道在tf.contrib.learn tutorial中重构代码的首选方法是什么,以删除tensorflow 0.12中的以下警告:

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不能使用自定义Estimator,而fit方法调用

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

会触发警告,即使x,y和batch_size未传递给fit方法。

2 个答案:

答案 0 :(得分:1)

你可以试着"包裹"全部在 SKCompat 中。 我有一个比喻问题,我已经部分解决了(例子)tf.contrib.learn.SKCompat(tf.contrib.learn.DNNClassifier( ...).fit(...)。 否则,您可以按照tensorflow.org

中的说明使用input_fn()

答案 1 :(得分:0)

我听说Tensorflow 1.0(将很快发布),它将从tf.learn移至tf.contrib.learn的主要发行版。如果你想尝试一下,那可能会有用。