警告:tensorflow:不推荐使用TensorFlowDNNRegressor类。请考虑使用DNNRegressor作为替代方案

时间:2016-06-10 19:08:54

标签: python tensorflow skflow

昨天,我已将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:使用ndim转换数组> 0到   index将导致将来result_shape.insert出错(dim,   1)

     

警告:tensorflow:不推荐使用TensorFlowDNNRegressor类。   请考虑使用DNNRegressor作为替代方案。

1 个答案:

答案 0 :(得分:1)

这只是一个警告,所以你的模型将训练好。 Skflow(或TFLearn同样的事情)建议您使用skflow.DNNRegressor代替skflow.TensorFlowDNNRegressor

以下是DNNRegressordoc

的论点
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,新功能还没有用,所以我建议你暂时保留旧功能!