python - estimator.predict テンソルの出力に関する問題

Question

私は現在 tf 1.4 を使用していますが、tf.contrib.factorization.KMeansClustering推定器の予測を調べるのに助けが必要です。現在のコード セグメントは次のようになります。

km = KMeansClustering(num_clusters=8,initial_clusters=KMeansClustering.KMEANS_PLUS_PLUS_INIT,model_dir=MODEL,relative_tolerance=0.01)

result = km.train(input_fn=lambda: gen_input(body))

input_fn = tf.estimator.inputs.pandas_input_fn(x={'x':tst}, shuffle=False)

y = result.predict(input_fn)

body と tst はパンダのデータフレームです。print(y)与えます：

<generator object Estimator.predict at 0x11ebecba0>

print(list(y))そして、 を呼び出す、print(next(y))または y を反復するなど、検索したことを試してみます。

for i in y:
    ...

for i in y.items():
    ...

for i in enumerate(y):
    ...

など、エラーが発生しますTypeError: data must be either a numpy array or pandas DataFrame if pandas is installed; got dict。これをオンラインで試して印刷する他の方法が見つかりません。ありがとう