python - 警告 :tensorflow:TensorFlowDNNRegressor class is deprecated. 请考虑使用 DNNRegressor 作为替代方案

Question

昨天，我将 Tensorflow 从 0.8 更新到了 0.9 版本。之后，我在 Skflow 模型中观察到以下警告。谁能告诉我这是什么意思吗？有办法解决吗？是否有其他方法可以使用 Skflow 对 DNN 回归器进行建模？

/Library/Frameworks/Python.framework/Versions/3.5/lib/python3.5/site-packages/tensorflow/python/ops/array_ops.py:1197:VisibleDeprecationWarning: converting an array with ndim > 0 to anindex will result in an error in the future result_shape.insert(dim,1)

WARNING:tensorflow:TensorFlowDNNRegressor class is deprecated.Please consider using DNNRegressor as an alternative.

Answer 1

这只是一个警告，因此您的模型可以正常训练。 Skflow(或 TFLearn 相同的东西)建议您使用 skflow.DNNRegressor 而不是 skflow.TensorFlowDNNRegressor。

以下是 DNNRegressor 和 doc 的参数:

tf.contrib.learn.DNNClassifier.__init__(
  hidden_units,
  feature_columns=None,
  model_dir=None,
  n_classes=2,
  weight_column_name=None,
  optimizer=None,
  activation_fn=relu,
  dropout=None,
  config=None)

但是，根据this previous post ，新功能还无法使用，所以我建议您暂时使用旧功能!