我正在尝试使用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'>
任何人都可以解释这个问题吗?