Tensorflow中的预测DNNRegressor

时间:2016-11-21 05:39:35

标签: python tensorflow

我正在尝试使用Tensorflow的这个教程:https://www.tensorflow.org/versions/master/tutorials/input_fn/index.html#building-the-input-fn

定义回归量:

regressor = tf.contrib.learn.DNNRegressor(
        feature_columns=feature_cols, 
        hidden_units=[10,10]
)

训练:

regressor.fit(input_fn=lambda: input_fn(training_set), steps=5500)

我的问题来自下一步。它们是关于预测的结果。

预测:

教程提供的标准版本:

y = regressor.predict(input_fn=lambda: input_fn(prediction_set))
print ("Predictions: {}".format(str(y)))

但是输出并不是我所期望的,它给了我:

Predictions: <generator object _infer_model_as_iterable at 0x7f8134159aa0>

我还试图检查它的类型:

print type(y)

#return: <type 'generator'>

任何人都可以解释这个问题吗?

0 个答案:

没有答案