я обучил LinearRegressor с двумя функциями: х, у и этикеток: лKeyError в Tensorflow при вызове предсказать на обученной модели
def train_input_fn():
x = [1,2,3,4]
y = [2,3,4,5]
feature_cols = tf.constant(x)
labels = tf.constant(y)
return feature_cols, labels
x = tf.contrib.layers.real_valued_column("x")
y = tf.contrib.layers.real_valued_column("y")
m = tf.contrib.learn.LinearRegressor(feature_columns=[ x,y],
model_dir=model_dir)
m.fit(input_fn=train_input_fn, steps=100)
После тренировки я хочу, чтобы предсказать из двух новых значений
new_sample = np.array([20,20])
m.predict(new_sample)
, но я получаю сообщение об ошибке при вызове предсказать
File "/usr/local/lib/python2.7/dist-packages/tensorflow/contrib/layers/python/layers/feature_column.py", line 870, in insert_transformed_feature
input_tensor = columns_to_tensors[self.name]
KeyError: 'x'
кто-нибудь знает, почему я получаю KEYER ROR?
Привет Никлас, как решить эту проблему? У меня такая же ошибка, как и вы. И, pred = self.model.predict_proba (input_fn = lambda: self.input_fn (data)), он работает. Однако, когда я использую X в качестве входных данных, это не сработало. – user48135
Привет @ user48135 Я решил это с принятым ответом ниже – Niclas