张量流 - tf.图层 :函数 .fit 出错

Question

我正在运行cnn_mnist github上给出的示例，该示例正在使用图层模块。我可以运行该程序，但出现警告，告诉我其中一个函数已被弃用。我找不到需要使用哪个新功能。

WARNING:tensorflow:From <ipython-input-14-ee49e8b76469>:25: calling BaseEstimator.fit (from tensorflow.contrib.learn.python.learn.estimators.estimator) with batch_size is deprecated and will be removed after 2016-12-01. Instructions for updating: 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(...))

Answer 1

像这样修改该脚本就足够了：

# Create the Estimator
mnist_classifier = learn.SKCompat(learn.Estimator(
  model_fn=cnn_model_fn, model_dir="/tmp/mnist_convnet_model"))

但是，请打开 Github 问题以更新该示例，以免误导其他任何人。