我只是想知道在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方法。
答案 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
的主要发行版。如果你想尝试一下,那可能会有用。