tensorflow - KeyError 尽管存在列标题？

Question

我正在尝试使用 Tensorflow 训练线性回归器。

如果我通过自动确定实值列来实例化学习器，则拟合工作正常。

auto_feature_columns =
    tf.contrib.learn.infer_real_valued_columns_from_input(
        training_examples)
linear_regressor = tf.contrib.learn.LinearRegressor(
  feature_columns=auto_feature_columns,
  optimizer=tf.GradientDescentOptimizer(learning_rate=0.002),
  gradient_clip_norm=3.14
)

但是，当我尝试手动指定列时，出现 KeyErrors(对于所有 3 列):

linear_regressor = tf.contrib.learn.LinearRegressor(
  feature_columns=[
      tf.contrib.layers.real_valued_column('tailLength'),
      tf.contrib.layers.real_valued_column('whiskerLength'),
      tf.contrib.layers.real_valued_column('playfulness')
  ],
  optimizer=tf.GradientDescentOptimizer(learning_rate=0.002),
  gradient_clip_norm=3.14
)

这很奇怪，因为我确信这 3 个是我训练数据的特征。

Answer 1

当您尝试拟合估算器时，是否会发生 KeyError？

如果是这样，也许您应该使用输入函数，允许特征操作，它返回 feature_columns 和具有这些键的标签，如 https://www.tensorflow.org/tutorials/input_fn/

我在使用 estimator.fit(x, y) 时遇到了同样的错误...

干杯，

拉斐尔。